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 vote219 views1 answer
Pandas for column in dataframe
Python
3 votes226 views1 answer
Calculate percentile pandas dataframe
Python
0 vote251 views1 answer
Write a variable and assin a string to it
Python
0 vote242 views1 answer
Dimension of an indez pandas
Python
5 votes227 views3 answers
Debugging python
Python
2 votes240 views1 answer
Django flush database
Python Django
0 vote216 views1 answer
Python - concatenate if null
Python
20 votes222 views3 answers
String to float python
Python
0 vote221 views1 answer
Isapha
Python
0 vote241 views1 answer
Mumtiply to matrices python
Python