Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
-from transformers import pipeline
-
-model_id = "thrishala/mental_health_chatbot"
+from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
+import torch
 
 model_id = "thrishala/mental_health_chatbot"
 
@@ -49,6 +48,7 @@ def respond(
         print(f"Error during generation: {e}")
         yield "An error occurred during generation." #Handle generation errors.
 
+
 # 4. Gradio Interface (No changes needed here)
 demo = gr.ChatInterface(
     respond,