MesonWarrior commited on
Commit
b20189a
1 Parent(s): c199f8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -15,8 +15,8 @@ def generate_text(
15
  ):
16
  models_map = {
17
  "Мои любимые юморески": "gpt2-vk-aneki",
18
- "Калик)": "gpt2-vk-kalik",
19
- "бугро тред": "gpt2-vk-bugro"
20
  }
21
 
22
  model = "MesonWarrior/" + models_map[model_name]
@@ -37,7 +37,7 @@ def interface():
37
  with gr.Column():
38
  with gr.Row():
39
  model = gr.Dropdown(
40
- ["Мои любимые юморески", "Калик)", "бугро тред"],
41
  label="Модель (Текст какого паблика генерировать)",
42
  value="бугро тред",
43
  )
@@ -95,8 +95,8 @@ with gr.Blocks(
95
  Файнтюны модели [ai-forever/rugpt3medium_based_on_gpt2](https://huggingface.co/ai-forever/rugpt3medium_based_on_gpt2) по вашим любимым пабликам ВКонтакте.
96
  #### Паблики представленные в моделях:
97
  - Мои любимые юморески 🎩
98
- - Калик) 🍏🍎💨 <sub><sup>(Обучено на постах и комментариях)</sup></sub>
99
  - бугро тред 💥
 
100
  """)
101
  interface()
102
 
 
15
  ):
16
  models_map = {
17
  "Мои любимые юморески": "gpt2-vk-aneki",
18
+ "бугро тред": "gpt2-vk-bugro",
19
+ "Калик)": "gpt2-vk-kalik"
20
  }
21
 
22
  model = "MesonWarrior/" + models_map[model_name]
 
37
  with gr.Column():
38
  with gr.Row():
39
  model = gr.Dropdown(
40
+ ["Мои любимые юморески", "бугро тред", "Калик)"],
41
  label="Модель (Текст какого паблика генерировать)",
42
  value="бугро тред",
43
  )
 
95
  Файнтюны модели [ai-forever/rugpt3medium_based_on_gpt2](https://huggingface.co/ai-forever/rugpt3medium_based_on_gpt2) по вашим любимым пабликам ВКонтакте.
96
  #### Паблики представленные в моделях:
97
  - Мои любимые юморески 🎩
 
98
  - бугро тред 💥
99
+ - Калик) 🍏🍎💨 <sub><sup>(Обучено на спорном датасете из постов и комментариев, надо бы переобучить на данных получше)</sup></sub>
100
  """)
101
  interface()
102