Update app.py

app.py

@@ -23,15 +23,21 @@ def generate_advice(prompt, model, tokenizer):
     generated_advice = tokenizer.decode(output[0], skip_special_tokens=True)
     return generated_advice
 
-# Create a Gradio interface
+# Function to handle button click
+def on_button_click(prompt):
+    return generate_advice(prompt, fine_tuned_model, tokenizer)
+
+# Create a Gradio interface with a button
 iface = gr.Interface(
-    fn=lambda prompt: generate_advice(prompt, fine_tuned_model, tokenizer),
+    fn=on_button_click,
     inputs=gr.Textbox(),
     outputs=gr.Textbox(),
     live=True,
     title="Depression Advice Generator",
-    description="Enter your feelings or ask for advice, and the model will generate supportive text.",
+    description="Enter your feelings or ask for advice, and click the button to get supportive text.",
+    # Add a button
+    live_button="Generate Advice",
 )
 
 # Run the Gradio interface
-iface.launch()
+iface.launch()