File size: 2,623 Bytes
826e378
 
26e737a
 
 
 
 
826e378
42d31a1
a24b436
 
 
 
 
aa95c30
a24b436
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
fd0079f
a24b436
41b8015
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
---
license: apache-2.0
datasets:
- Den4ikAI/ru_sberquad_long_answers
language:
- ru
pipeline_tag: text2text-generation
---
# Встроенный инференс генерирует плохо. Запускайте модель локально
T5 для генерации пары вопрос-ответ на русском языке.
Использование:
```python
import torch
from transformers import AutoModelForSeq2SeqLM, T5Tokenizer
model_name = "PyWebSol/QA-Gen"
tokenizer = T5Tokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

def generate(text, **kwargs):
    inputs = tokenizer(text, return_tensors='pt')   
    with torch.no_grad():
        hypotheses = model.generate(**inputs, num_beams=1, **kwargs, max_new_tokens=512)
        print(hypotheses)
    return tokenizer.decode(hypotheses[0], skip_special_tokens=True)

qa = generate(
        "К особым префектурам можно отнести Токио, Киото, Осаку и Хоккайдо. В период Эдо (1603—1867), сёгунат установил 9 городских районов, которыми управляли чиновники из центра (奉行支配地), и 302 районных города, которыми управляли городские чиновники (郡代支配地). С наступлением эпохи Мэйдзи 9 городских центров были превращены в округа фу, а 302 районных города — в префектуры кэн. В 1871 г., после административной реформы, в Японии было установлено 3 городских префектур фу — Токио, Киото и Осака. В 1943 г. городская префектура Токио была переименована в столицу то (хотя закона о столице утверждено не было)."
).split(" <|split|> ")

question, answer = qa

print(f"Вопрос: {question}")
print(f"Ответ: {answer}")
# Вопрос: В каком период было установлено 3 городских префектур фу — Токио, Киото и Осака?
# Ответ: В 1871 году.
```

Модель может быть полезна для автоматизированной генерации наборов данных по тексту для обучения других NLP моделей.

Телеграм бот для фильтрации спама в чатах: https://t.me/RuModeratorAI_Bot