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

2 votes162 views1 answer
Pandas remame a no name column
Python
0 vote171 views1 answer
How to find permutation of numbers in python
Python
4 votes178 views1 answer
Tkinter change label text color
Python
0 vote171 views1 answer
Edit models in django admin
Python Django
0 vote151 views1 answer
Python import multiple csv
Python
2 votes168 views1 answer
Colab drive access
Python
0 vote163 views1 answer
Pandas filter dataframe if an elemnt is in alist
Python
0 vote145 views1 answer
Sin , Cos Graph using python turtle.
Python
0 vote174 views1 answer
How to read panda column
Python
3 votes177 views1 answer
Python split on first occurrence
Python