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 vote288 views1 answer
Python get packages path
Python
0 vote268 views1 answer
Python array remove new element
Python
0 vote311 views1 answer
Column names reading csv file python
Python
0 vote328 views1 answer
Python dash bootstrap buttons with icons
Python
0 vote303 views1 answer
Kmpm]pomfyukruk6nfgngnzgnzggngnxfgnfgxfgfgxfggnxfggngnggngngngngngngngn. ';';';';'; ; ; ; ;; ; ; ; ; ; ; ; ;
Python
0 vote275 views1 answer
Merge the tool hacker rank solution
Python
4 votes309 views1 answer
Df select first n rows
Python
14 votes285 views2 answers
How to convert csv to excel in python
Python
0 vote303 views1 answer
Python reduce function to sum array
Python
0 vote292 views1 answer
Save artist animation puython
Python