Den4ikAI's picture
Update README.md
05c8a6b
|
raw
history blame
792 Bytes
---
license: mit
pipeline_tag: question-answering
widget:
- context: "Пушкин родился 6 июля 1799 года"
- text: "Когда родился Пушкин?"
example_title: "test"
---
обученный rubert от sberbank-ai/ruBert-base.
размер выборки - 4.
Эпохи - 4.
```python
from transformers import pipeline
qa_pipeline = pipeline(
"question-answering",
model="Den4ikAI/rubert_large_squad_2",
tokenizer="Den4ikAI/rubert_large_squad_2"
)
predictions = qa_pipeline({
'context': "Пушкин родился 6 июля 1799 года",
'question': "Когда родился Пушкин?"
})
print(predictions)
# output:
#{'score': 0.8013797664642334, 'start': 15, 'end': 31, 'answer': '6 июля 1799 года'}
```