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

10 votes160 views4 answers
Set method in python
Python
0 vote154 views1 answer
How to write a table from 1 to 10 with for loop in fython in 3 lines
Python
4 votes205 views1 answer
Python 3.7 download for windows 7 32-bit
Python
7 votes150 views2 answers
Python concatenate list of lists
Python
0 vote197 views1 answer
Django extract array from post matrix
Python Django
0 vote154 views1 answer
Tf.io path copy
Python
31 votes154 views3 answers
How to adda vaslues to data frame
Python
3 votes169 views1 answer
How to open webcam with python
Python
1 vote142 views1 answer
Python anagram finder
Python
5 votes143 views3 answers
Print command python
Python