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

3 votes306 views1 answer
Time decorator python
Python
4 votes361 views1 answer
What is cleaned data in django
Python
3 votes360 views6 answers
Counter in python
Python
0 vote334 views1 answer
Jupyter upload folder
Python
2 votes341 views1 answer
How to find geometric mean in python
Python
0 vote364 views1 answer
Python garbaze collection
Python
0 vote350 views1 answer
Pandas.describe per group
Python
0 vote365 views1 answer
Reset Index & Retain Old Index as Column in pandas
Python
9 votes370 views1 answer
How to create a keylogger in python
Python
0 vote383 views1 answer
R sample() funciton in python
Python