Update app.py

app.py

@@ -4,16 +4,16 @@ import streamlit as st
 from transformers import pipeline
 
 # Function to load Hugging Face models
-def load_model(model_name):
-    return pipeline("text-generation", model=model_name, device=0)  # Adjust the model type accordingly
+def load_model():
+    return pipeline("conversational", model="alpindale/goliath-120b")
 
 # Page 1: Welcome and Chatbot
 def page_welcome():
     st.title("Welcome to Your Virtual Therapist")
     st.write("Feel free to chat with our virtual therapist!")
 
-    # Load your preferred Hugging Face chatbot model
-    chatbot_model = load_model("your-chatbot-model-name")
+    # Load the provided Hugging Face chatbot model
+    chatbot_model = load_model()
 
     user_input = st.text_input("You: ")
     if user_input: