Updated system prompt

app.py

@@ -9,7 +9,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 def chat(message, history):
     # 1. Format the conversation using the official template
-    messages = [{"role": "user", "content": message}]
+    messages = [{"role": "user", "content": message},
+               {"role": "system", "content": "Use context if relevant, otherwise answer generally."},]
     prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
     
     # 2. Generate with a 'stop' token