Update app.py

app.py

@@ -43,16 +43,11 @@ st.markdown(
 # Add the blurred background div
 st.markdown('<div class="blurred-background"></div>', unsafe_allow_html=True)
 
-# Load the fine-tuned model and tokenizer
 @st.cache_resource
-def load_model_and_tokenizer():
-    config = PeftConfig.from_pretrained("zementalist/llama-3-8B-chat-psychotherapist")
-    base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
-    model = PeftModel.from_pretrained(base_model, "zementalist/llama-3-8B-chat-psychotherapist")
-    tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
-    return model, tokenizer
+def load_pipeline():
+    return pipeline("text-generation", model="QuantFactory/Mental-Health-FineTuned-Mistral-7B-Instruct-v0.2-GGUF")
 
-model, tokenizer = load_model_and_tokenizer()
+text_generation_pipe = load_pipeline()
 
 # Load dataset for reference (optional)
 @st.cache_resource