Update app.py

app.py

@@ -27,16 +27,14 @@ def generate_advice(prompt):
     advice = dataset["train"][predicted_label_id]["text"]
     return advice
 
-# Create a Gradio interface
+# Create the Gradio interface
 iface = gr.Interface(
-    fn=generate_answer,
-    inputs=[
-        gr.Textbox(placeholder="Enter context", default=contexts[0]),
-        gr.Textbox(placeholder="Enter question", default=questions[0]),
-        gr.Button("Generate Answer"),
-    ],
-    outputs=gr.Textbox(placeholder="Answer"),
-    live=True
+    fn=generate_advice,
+    inputs=gr.Textbox(),
+    outputs=gr.Textbox(),
+    live=True,
+    title="Depression Advice Generator",
+    description="Enter your feelings, and get supportive advice generated by a fine-tuned BERT model.",
 )
 
 # Launch the Gradio interface