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 votes171 views1 answer
Flask form
Python Flask
0 vote176 views2 answers
Pyqt close window
Python
-3 votes188 views1 answer
Django socketio web chat example
Python Django
3 votes218 views1 answer
Django group by
Python Django
0 vote168 views1 answer
Set xlim histogram python
Python
53 votes198 views2 answers
Add values to dictionary key python
Python
1 vote161 views1 answer
Python read integer from stdin
Python
0 vote166 views1 answer
Django import excel file from same directory
Python
0 vote220 views1 answer
Add to middle of list python
Python
0 vote160 views1 answer
Multiclass.roc plot title
Python