Upload app.py

app.py

@@ -5,55 +5,51 @@ from huggingface_hub import login
 login(token="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")
 
 def generate_text(
-    model,
+    model_name,
     text,
     min_length,
     max_length
-#    do_not_truncate
 ):
+    models_map = {
+        "Юморески": "gpt2-vk-aneki",
+        "Калик": "gpt2-vk-kalik",
+        "Бугро": "gpt2-vk-bugro"
+    }
+
+    model = "MesonWarrior/" + models_map[model_name]
+
     pipe = pipeline(
         'text-generation',
-        model='MesonWarrior/gpt2-vk-bugro',
-        tokenizer='MesonWarrior/gpt2-vk-bugro',
+        model=model,
+        tokenizer=model,
         min_length=min_length,
         max_length=max_length,
-#        do_not_truncate=do_not_truncate,
         use_auth_token=True
     )
 
-    print('generating...')
-
-    output = pipe(text)
-
-    print(output)
-
-    return output[0]['generated_text']
+    return pipe(text)[0]['generated_text']
 
 def interface():
     with gr.Row():
         with gr.Column():
             with gr.Row():
                 model = gr.Dropdown(
-                    ["Бугро", "Юморески", "Калик"], label="Model", value="Бугро",
+                    ["Юморески", "Калик", "Бугро"], label="Модель (Текст какого паблика генерировать)", value="Бугро",
                 )
-            text = gr.Textbox(lines=7, label="Input text")
-        output = gr.Textbox(lines=12, label="Output text")
+            text = gr.Textbox(lines=7, label="Входной текст", placeholder="Введите текст который продолжит нейросеть...")
+        output = gr.Textbox(lines=12, label="Выходной текст", placeholder="Здесь будет текст сгенерированный нейросетью...")
     with gr.Row():
         with gr.Column():
             min_length = gr.Slider(
                 minimum=0, maximum=128, value=32, step=1,
                 label="Min Length",
+                info="Минимальное количество символов в выходном тексте."
             )
-
             max_length = gr.Slider(
-                minimum=0, maximum=512, value=96, step=1,
+                minimum=0, maximum=512, value=64, step=1,
                 label="Max Length",
+                info="Максимальное количество символов в выходном тексте."
             )
-
-#            do_not_truncate = gr.Checkbox(
-#                True,
-#                label="Do not truncate"
-#            )
         with gr.Column():
             with gr.Row():
                 generate_btn = gr.Button(
@@ -67,7 +63,6 @@ def interface():
                 text,
                 min_length,
                 max_length,
-#                do_not_truncate
             ],
             outputs=output,
         )
@@ -77,6 +72,10 @@ with gr.Blocks(
         gr.Markdown("""
         ## GPT2 VK
         Файнтюны модели [ai-forever/rugpt3medium_based_on_gpt2](https://huggingface.co/ai-forever/rugpt3medium_based_on_gpt2) по вашим любимым пабликам ВКонтакте.
+        Паблики представленные в моделях:
+        - Мои любимые юморески 🎩
+        - Калик) 💨
+        - бугро тред 🅰
         """)
         interface()