Update app.py

app.py

@@ -25,8 +25,7 @@ tokenizer.eos_token_id = eos_token_id
 model.config.eos_token_id = eos_token_id
 
 def format_prompt(user_message, system_message="You are YiTonic, an AI language model created by Tonic-AI. You are a cautious assistant. You carefully follow instructions. You are helpful and harmless and follow ethical guidelines and promote positive behavior."):
-    prompt_dict = {"role": "user", "content": f"{user_message}{system_message}"}
-    prompt = str(prompt_dict)
+    prompt = f"<|im_start|>assistant\n{system_message}<|im_end|>\n<|im_start|>\nuser\n{user_message}<|im_end|>\nassistant\n"
     return prompt
 
 def predict(message, system_message, max_new_tokens=4056, temperature=3.5, top_p=0.9, top_k=40, model_max_length = 32000, do_sample=False):