Update app.py

app.py

@@ -11,7 +11,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model = model.to(device=device)
 
 def format_prompt(message, history):
-    prompt = ""
+    prompt = "<|system|>\nYou are TinyLlama, a friendly AI assistant.</s>"
     for user_prompt, bot_response in history:
         prompt += f"\n<|user|>\n{user_prompt}</s>"
         prompt += f"\n<|assistant|>\n{bot_response}</s>"