Update app.py

app.py

@@ -9,24 +9,25 @@ except subprocess.CalledProcessError as e:
     print("Error:", e)
     print(e.stderr)
 
-def chatbot(prompt):
-    command = ["./run", "stories15M.bin", "-t", "0.8", "-p", "0.9", "-n", "256", "-i", f"{prompt}"]
+def chatbot(prompt, temperature, topt, maxtoken, model_file):
+    command = ["./run", model_file, "-t", str(temperature), "-p", str(topt), "-n", str(maxtoken), "-i", f"{prompt}"]
     try:
         result = subprocess.run(command, capture_output=True, text=True, check=True, shell=False)
         response = result.stdout
     except subprocess.CalledProcessError as e:
         response = "Error occurred while processing the request."
     return response
-    
-
 
 with gr.Blocks() as demo:
     gr.Markdown("HF Spaces for TinyStories")
     with gr.Row():
         inp = gr.Textbox(placeholder="Type the beginning of the story")
+        temperature_slider = gr.Slider(minimum=0.1, maximum=2.0, default=0.8, label="Temperature")
+        topt_slider = gr.Slider(minimum=0.1, maximum=1.0, default=0.9, label="Topt")
+        maxtoken_slider = gr.Slider(minimum=16, maximum=512, default=256, label="Max Tokens")
+        model_file_dropdown = gr.Dropdown(choices=["stories15M.bin", "stories42M.bin", "stories110M.bin"], label="Model File")
         out = gr.Textbox()
     btn = gr.Button("Run")
-    btn.click(fn=chatbot, inputs=inp, outputs=out)
+    btn.click(fn=chatbot, inputs=[inp, temperature_slider, topt_slider, maxtoken_slider, model_file_dropdown], outputs=out)
 
 demo.launch()
-