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

1 vote192 views1 answer
Not getting spanish characters python
Python
32 votes190 views4 answers
Python pip install matplotlib
Python
0 vote212 views1 answer
MULTIPLE Model Search in Django rest framework
Python
0 vote212 views2 answers
Find location of a class in python
Python
-1 vote186 views1 answer
Usage code grepper
Python
0 vote245 views1 answer
Import tsv as dataframe python
Python
5 votes206 views3 answers
How to print multiple strings on one line in python
Python
0 vote334 views1 answer
Python string: .split()
Python
0 vote247 views1 answer
How to import flask restful using pip
Python Flask
0 vote207 views1 answer
List addition within a list comprehension
Python