Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 #from transformers import AutoTokenizer
-from llama_cpp import Llama
+#from llama_cpp import Llama
 from transformers import pipeline
 #from peft import PeftModel, PeftConfig
 #from transformers import AutoModelForCausalLM
@@ -51,7 +51,7 @@ st.markdown('<div class="blurred-background"></div>', unsafe_allow_html=True)
 @st.cache_resource
 def load_text_generation_model():
     try:
-        return pipeline("text-generation", model="thrishala/mental_health_chatbot")
+        return pipeline("text-generation", model="google/gemma-2-9b-it")
     except Exception as e:
         st.error(f"Error loading model: {e}")
         return None
@@ -69,7 +69,7 @@ dataset = load_counseling_dataset()
 st.title("Mental Health Counseling Chat")
 st.markdown("""
 Welcome to the **Mental Health Counseling Chat Application**.  
-This platform is designed to provide **supportive, positive, and encouraging responses** based on mental health counseling expertise.
+This platform is designed to provide **supportive, positive, and encouraging responses** using a fast and efficient language model.
 """)
 
 # Check if the model loaded correctly
@@ -95,7 +95,7 @@ else:
                 response = text_generator(prompt, max_length=200, num_return_sequences=1)
                 
                 # Extract and display the response
-                counselor_reply = response[0]["generated_text"].split("Counselor:")[-1].strip()
+                counselor_reply = response[0]["generated_text"].strip()
                 st.subheader("Counselor's Response:")
                 st.write(counselor_reply)
             except Exception as e:
@@ -111,6 +111,4 @@ else:
     - [National Suicide Prevention Lifeline](https://suicidepreventionlifeline.org)
     """)
 
-    st.sidebar.info("This application is not a replacement for professional counseling. If you are in crisis, please seek professional help immediately.")
-
-
+    st.sidebar.info("This application is not a replacement for professional counseling. If you are in crisis, please seek professional help immediately.")