File size: 2,772 Bytes
1d19835
 
939b03a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1d19835
37218c3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
---
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.