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

8 votes174 views3 answers
Python append csv to dataframe
Python
2 votes179 views1 answer
How to redirect in flask to the same page
Python Flask
10 votes146 views3 answers
Python click on screen
Python
8 votes130 views2 answers
How to combine two arrays in python
Python
0 vote154 views1 answer
Unable to get local issuer certificate python
Python
0 vote145 views1 answer
How to fill an array with consecutive numbers python
Python
0 vote146 views1 answer
Root value of a column pandas
Python
20 votes147 views5 answers
Remove first character from string python
Python
0 vote138 views1 answer
Class indexing
Python
13 votes155 views5 answers
How to check if a input is an integer python
Python