Views
No views yet
1from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
2
3# Load model & tokenizer
4deberta_model = AutoModelForQuestionAnswering.from_pretrained('navteca/deberta-v3-large-squad2')
5deberta_tokenizer = AutoTokenizer.from_pretrained('navteca/deberta-v3-large-squad2')
6
7# Get predictions
8nlp = pipeline('question-answering', model=deberta_model, tokenizer=deberta_tokenizer)
9
10result = nlp({
11 'question': 'How many people live in Berlin?',
12 'context': 'Berlin had a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers.'
13})
14
15print(result)
16
17#{
18# "answer": "3,520,031"
19# "end": 36,
20# "score": 0.96186668,
21# "start": 27,
22#}