Update app.py

app.py

@@ -8,10 +8,13 @@ def generate_text(
     model_name,
     text,
     min_length,
-    max_length
+    max_length,
+    temperature,
+    top_k,
+    top_p
 ):
     models_map = {
-        "Мои любимые юмореск": "gpt2-vk-aneki",
+        "Мои любимые юморески": "gpt2-vk-aneki",
         "Калик)": "gpt2-vk-kalik",
         "бугро тред": "gpt2-vk-bugro"
     }
@@ -27,7 +30,7 @@ def generate_text(
         use_auth_token=True
     )
 
-    return pipe(text)[0]['generated_text']
+    return pipe(text, temperature=temperature, top_k=top_k, top_p=top_p)[0]['generated_text']
 
 def interface():
     with gr.Row():
@@ -52,6 +55,19 @@ def interface():
                 label="Максимальная длина",
                 info="Максимальное количество символов в выходном тексте."
             )
+            temperature = gr.Slider(
+                minimum=0.05, maximum=1.95, value=0.9, step=0.05,
+                label="Температура",
+                info="Чем выше тем рандомнее, чем ниже тем больше повторений."
+            )
+            top_k = gr.Slider(
+                minimum=0, maximum=10, value=100, step=0.05,
+                label="Top K",
+            )
+            top_p = gr.Slider(
+                minimum=0, maximum=1, value=0.9, step=0.05,
+                label="Top P",
+            )
         with gr.Column():
             with gr.Row():
                 generate_btn = gr.Button(
@@ -65,6 +81,9 @@ def interface():
                 text,
                 min_length,
                 max_length,
+                temperature,
+                top_k,
+                top_p
             ],
             outputs=output,
         )