Rubert Tiny Squad

🚀 Den4ikAI/rubert-tiny-squad

A question-answering model fine-tuned on rubert from cointegrated/rubert-tiny2.

🚀 Quick Start

This model is fine-tuned on rubert from cointegrated/rubert-tiny2. It has a dataset size of 4 and was trained for 16 epochs.

from transformers import pipeline
qa_pipeline = pipeline(
    "question-answering",
    model="Den4ikAI/rubert-tiny-squad",
    tokenizer="Den4ikAI/rubert-tiny-squad"
)
predictions = qa_pipeline({
    'context': "Пушкин родился 6 июля 1799 года",
    'question': "Когда родился Пушкин?"
})
print(predictions)
# output:
#{'score': 0.9413797664642334, 'start': 15, 'end': 31, 'answer': '6 июля 1799 года'}

📄 License

This project is licensed under the MIT license.