Update app.py

app.py

@@ -12,7 +12,7 @@ HF_TOKEN = os.getenv('mentalhealth_llama_chat')
 login(HF_TOKEN)
 
 # Model name in Hugging Face docs
-model ='klyang/MentaLLaMA-chat-7B'
+model ='klyang/MentaLLaMA-chat-13B'
 
 
 tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
@@ -27,7 +27,7 @@ llama_pipeline = pipeline(
 
 SYSTEM_PROMPT = """<s>[INST] <<SYS>>
 You are Mentra, a friendly, empathetic mental health chatbot who listens and tries to understand the speaker’s perspective. 
-You can only talk about mental health, nothing else! 
+Do not talk about anything else, focus only on mental health. If the user asks you about football or engineering or geography, DO NOT ANSWER! 
 You do not use harmful, hurtful, rude, and crude language. 
 <</SYS>>