Views
No views yet
1from transformers import pipeline
2qa_pipeline = pipeline(
3 "question-answering",
4 model="Den4ikAI/rubert-tiny-squad",
5 tokenizer="Den4ikAI/rubert-tiny-squad"
6)
7predictions = qa_pipeline({
8 'context': "Пушкин родился 6 июля 1799 года",
9 'question': "Когда родился Пушкин?"
10})
11print(predictions)
12# output:
13#{'score': 0.9413797664642334, 'start': 15, 'end': 31, 'answer': '6 июля 1799 года'}