Views
No views yet
1from transformers import AutoTokenizer, T5ForConditionalGeneration
2
3model_name = "wanderer-msk/ruT5-base_headline_generation"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = T5ForConditionalGeneration.from_pretrained(model_name)
6
7news_text = """Земляне продолжают осваивать Марс.
8Колонисты уже посадили на красной планете 42 яблони."""
9
10model_input = tokenizer(
11 news_text,
12 truncation=True,
13 max_length=1024,
14 return_tensors="pt"
15)
16model_output = model.generate(model_input["input_ids"])
17news_headline = tokenizer.decode(
18 model_output.squeeze(),
19 skip_special_tokens=True
20)
21
22print(news_headline)