sarahai's picture
Update README.md
37218c3 verified
|
raw
history blame
2.77 kB
---
license: apache-2.0
library_name: transformers
datasets:
- IlyaGusev/gazeta
language:
- ru
pipeline_tag: summarization
tags:
- summarization
- summarizer
- суммаризатор
- text-generation-inference
widget:
- text: >-
В одном из зоопарков Петербурга пингвин по имени Гектор решил, что пора искать
себе пару и отправился в авантюрное путешествие прямо из своего вольера.
Сотрудники зоопарка утром обнаружили пропажу и начали поиски, но Гектор уже
прошел через весь город, пытаясь даже заказать кофе на Невском проспекте.
Приключения пингвина вызвали восторг у жителей города, которые делились фото и
видео в соцсетях с комментариями от "Может, ему просто надоела рыба?" до "Гектор
ищет любовь, давайте поможем ему!". В конце концов, Гектора нашли и вернули в
зоопарк, где ему обещали подыскать подругу. Эта история не только развеселила
горожан, но и напомнила о том, что даже у пингвинов есть сердца, способные на
большую любовь, а Гектор стал настоящей звездой, символом неутомимого стремления
к своей мечте.
example_title: Summarization Example 1
---
Russian text summarizer was fine-tuned from ai-forever/ruT5-base model and trained on ~60k rows samples' dataset.
Example Usage:
```python
model_name = "sarahai/ruT5-base-summarizer"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
device = torch.device("cpu") #if you are using cpu
input_text = "текст на русском" #your input in russian
input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to(device)
outputs = model.generate(input_ids, max_length=100, min_length=50, length_penalty=2.0, num_beams=4, early_stopping=True) #change according to your preferences
summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(summary)
```
References
Hugging Face Model Hub
T5 Paper
Disclaimer: The model's performance may be influenced by the quality and representativeness of the data it was fine-tuned on. Users are encouraged to assess the model's suitability for their specific applications and datasets.