Update app.py

app.py

@@ -77,8 +77,7 @@ def stream_chat(
     conversation.append({"role": "user", "content": message})
 
     
-    input_text = tokenizer.apply_chat_template(conversation, tokenize=False)
-    input_text += "<|im_start|>assistant\n"
+    input_text = tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt = True)
         
     inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)
     streamer = TextIteratorStreamer(tokenizer, timeout=60.0, skip_prompt=True, skip_special_tokens=True)