Update app.py

app.py

@@ -1,4 +1,5 @@
 import torch
+import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 device = "cuda" if torch.cuda.is_available() else "cpu"  # Automatically detect GPU or CPU
@@ -43,14 +44,19 @@ def generate_response(user_input):
     return chatbot_response
 
 # Continuous conversation loop
-while True:
+'''while True:
     user_input = input("You: ")  # Take user input
     if user_input.lower() in ["exit", "quit", "stop"]:
         print("Chatbot: Goodbye!")
         break
 
     response = generate_response(user_input)
-    print("Chatbot:", response)
+    print("Chatbot:", response)'''
+
+
+# Initialize the ChatInterface
+chatbot = gr.ChatInterface(fn=generate_response, title="Mental Health Chatbot")
+chatbot.launch()
     
     
 '''