jesthy commited on
Commit
1680f2d
·
verified ·
1 Parent(s): 5d5280f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -11
app.py CHANGED
@@ -1,19 +1,19 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Загружаем модель ruGPT‑3.5‑13B
5
  generator = pipeline(
6
  "text-generation",
7
- model="ai-forever/ruGPT-3.5-13B",
8
- device=0, # или -1, если без GPU
9
  torch_dtype="auto"
10
  )
11
 
12
- # Стили текстов
13
  styles = {
14
  "Дружелюбно": "в дружелюбном стиле",
15
  "Дерзко": "в дерзком стиле",
16
- "По‑мотивирующему": "в мотивационном стиле",
17
  "С юмором": "с юмором",
18
  "Философски": "в философском стиле"
19
  }
@@ -25,19 +25,18 @@ def clean_text(text):
25
  def generate_post(topic, style):
26
  style_text = styles.get(style, "")
27
  prompt = (
28
- # Чётко задаем роль, тему, стиль и формат
29
  "Ты — SMM‑копирайтер. "
30
  f"Тема: '{topic}'. Стиль: {style_text}. "
31
- "Сгенерируй **один короткий пост**, который вызывает эмоцию или мысль. "
32
- "Без советов, инструкций и рекомендаций."
33
  )
34
 
35
  results = generator(
36
  prompt,
37
- max_new_tokens=80, # сокращаем длину для лаконичности
38
  do_sample=True,
39
- temperature=0.4, # ниже — для более точного контроля
40
- top_p=0.9, # ограничиваем выбор токенов
41
  num_return_sequences=1
42
  )
43
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Используем лёгкую инструктивную модель
5
  generator = pipeline(
6
  "text-generation",
7
+ model="AlexWortega/instruct_rugptSmall",
8
+ device=0, # GPU, если доступен
9
  torch_dtype="auto"
10
  )
11
 
12
+ # Стили постов
13
  styles = {
14
  "Дружелюбно": "в дружелюбном стиле",
15
  "Дерзко": "в дерзком стиле",
16
+ "По‑мотивирующим": "в мотивационном стиле",
17
  "С юмором": "с юмором",
18
  "Философски": "в философском стиле"
19
  }
 
25
  def generate_post(topic, style):
26
  style_text = styles.get(style, "")
27
  prompt = (
 
28
  "Ты — SMM‑копирайтер. "
29
  f"Тема: '{topic}'. Стиль: {style_text}. "
30
+ "Сгенерируй один короткий пост, наполненный эмоцией или мыслью, "
31
+ "без советов, инструкций или рекомендаций."
32
  )
33
 
34
  results = generator(
35
  prompt,
36
+ max_new_tokens=80,
37
  do_sample=True,
38
+ temperature=0.4,
39
+ top_p=0.9,
40
  num_return_sequences=1
41
  )
42