Albert pretrained example

from transformers import AlbertTokenizer, AlbertForQuestionAnswering
import torch

tokenizer = AlbertTokenizer.from_pretrained('albert-base-v2')
model = AlbertForQuestionAnswering.from_pretrained('albert-base-v2')
question, text = "Who was Jim Henson?", "Jim Henson was a nice puppet"
input_dict = tokenizer.encode_plus(question, text, return_tensors='pt')
start_scores, end_scores = model(**input_dict)

Source: huggingface.co

Similar codes

0 vote303 views1 answer
Not en python
Python
0 vote300 views1 answer
Settings urls
Python
34 votes297 views6 answers
Python replace string
Python
3 votes295 views2 answers
String of numbers to list of integers python
Python
1 vote349 views1 answer
Sort a dict by values
Python
1 vote347 views1 answer
Factorial sequence code in python with while loops
Python
133 votes289 views6 answers
Python input
Python
27 votes331 views4 answers
Random letter generator python
Python
0 vote286 views1 answer
My name is raghuveer
Python
3 votes275 views1 answer
Python 3.9 features
Python