Update app.py

app.py

@@ -9,16 +9,24 @@ except subprocess.CalledProcessError as e:
     print("Error:", e)
     print(e.stderr)
 
-
 def chatbot(prompt):
-    import subprocess
-    command = ["./run", "stories15M.bin", "-t", "0.8", "-p", "0.9", "-n", "256", "-i", f'"{prompt}"']
+    command = ["./run", "stories15M.bin", "-t", "0.8", "-p", "0.9", "-n", "256", "-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")
+        out = gr.Textbox()
+    btn = gr.Button("Run")
+    btn.click(fn=chatbot, inputs=inp, outputs=out)
+
+demo.launch()
 
-iface = gr.Interface(fn=chatbot, inputs="text", outputs="text", live=True)
-iface.launch()