Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Загрузка модели и токенизатора
4model_name = "kremlinai/kremlinai-2"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(model_name)
7
8# Генерация комментария
9def generate_comment(news_title, max_length=100):
10 prompt = f"Заголовок: {news_title}\nКомментарий кремлебота:"
11 inputs = tokenizer(prompt, return_tensors="pt")
12 outputs = model.generate(
13 inputs.input_ids,
14 max_length=max_length,
15 do_sample=True,
16 top_p=0.9,
17 temperature=0.8
18 )
19 return tokenizer.decode(outputs[0], skip_special_tokens=True)
20
21# Пример вызова
22news_title = "Запад усиливает санкции против России"
23print(generate_comment(news_title))