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

4 votes266 views1 answer
Django pandas queryset
Python Django
0 vote236 views1 answer
Int to alphabet letter python
Python
0 vote254 views1 answer
How to find the medium, first, second and third quartile in a pandas data frame
Python
78 votes263 views6 answers
Save dataframe as csv
Python
0 vote272 views1 answer
India states django choices
Python
0 vote218 views1 answer
Python read in integers separated by spaces
Python
0 vote249 views2 answers
How to get every character in a string python
Python
20 votes237 views4 answers
How to make computer go in sleep mode using pythn
Python
0 vote249 views1 answer
Long to_bytes python how to use it
Python
1 vote228 views1 answer
Pandas percent change
Python