t

app.py

@@ -18,6 +18,9 @@ import gradio as gr
 # except Exception as e:
 #     ex = e
 
+temperature = gr.inputs.Slider(minimum=0, maximum=1.5, step=0.1, default=0.9, label="temperature")
+top_p = gr.inputs.Slider(minimum=0, maximum=1.0, step=0.1, default=0.9, label="top_p")
+
 import requests
 def f(text):
     try:
@@ -27,8 +30,8 @@ def f(text):
         payload = {
             "context": context,
             "token_max_length": 512,
-            "temperature": 1.0,
-            "top_p": 0.9,
+            "temperature": temperature,
+            "top_p": top_p,
         }
         response = requests.post("http://api.vicgalle.net:5000/generate", params=payload).json()
 
@@ -36,7 +39,6 @@ def f(text):
         return "error: \n"+str(e)
     return response['text']
 
-   
 
 iface = gr.Interface(fn=f, inputs="text", outputs="text")
 iface.launch(enable_queue=True)