Upload app.py

app.py

@@ -11,9 +11,9 @@ def generate_text(
     max_length
 ):
     models_map = {
-        "Юморески": "gpt2-vk-aneki",
-        "Калик": "gpt2-vk-kalik",
-        "Бугро": "gpt2-vk-bugro"
+        "Мои любимые юмореск": "gpt2-vk-aneki",
+        "Калик)": "gpt2-vk-kalik",
+        "бугро тред": "gpt2-vk-bugro"
     }
 
     model = "MesonWarrior/" + models_map[model_name]
@@ -34,7 +34,9 @@ def interface():
         with gr.Column():
             with gr.Row():
                 model = gr.Dropdown(
-                    ["Юморески", "Калик", "Бугро"], label="Модель (Текст какого паблика генерировать)", value="Бугро",
+                    ["Мои любимые юморески", "Калик)", "бугро тред"],
+                    label="Модель (Текст какого паблика генерировать)",
+                    value="бугро тред",
                 )
             text = gr.Textbox(lines=7, label="Входной текст", placeholder="Введите текст который продолжит нейросеть...")
         output = gr.Textbox(lines=12, label="Выходной текст", placeholder="Здесь будет текст сгенерированный нейросетью...")
@@ -42,18 +44,18 @@ def interface():
         with gr.Column():
             min_length = gr.Slider(
                 minimum=0, maximum=128, value=32, step=1,
-                label="Min Length",
+                label="Минимальная длина",
                 info="Минимальное количество символов в выходном тексте."
             )
             max_length = gr.Slider(
                 minimum=0, maximum=512, value=64, step=1,
-                label="Max Length",
+                label="Максимальная длина",
                 info="Максимальное количество символов в выходном тексте."
             )
         with gr.Column():
             with gr.Row():
                 generate_btn = gr.Button(
-                    "Generate", variant="primary", label="Generate",
+                    "Сгенерировать", variant="primary", label="Generate",
                 )
 
         generate_btn.click(