Update app.py

app.py

@@ -22,12 +22,16 @@ def generate_text(question):
     response = tokenizer.decode(tokens[0], skip_special_tokens=True)
     return response.split('\nA: ')[-1] 
 
-iface = gr.Interface(
-    fn=generate_text,
-    gr.Markdown(title),
-    gr.Markdown(description),
-    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
-    outputs="text"
-)
+# Gradio Blocks interface
+with gr.Blocks() as iface:
+    gr.Markdown(title)
+    gr.Markdown(description)
+    with gr.Row():
+        question = gr.Textbox(label="Enter your question", lines=2, placeholder="Enter your prompt here...")
+        submit_button = gr.Button("Submit")
+    output = gr.Textbox(label="Model Response")
 
+    submit_button.click(fn=generate_text, inputs=question, outputs=output)
+
+# Launch the interface
 iface.launch()