Update app.py

app.py

@@ -26,13 +26,10 @@ def generate(
         tokenizer.eos_token = "<|im_end|>"
         print(tokenizer)
         pipe.tokenizer = tokenizer
-        if model_name == "Locutusque/TinyMistral-248M-v3":
-            prompt = " ".join(history) + message
-        else:
-            prompt = f"<|im_start|>system\n{system}<|im_end|>\n"
-            for (user_turn, assistant_turn) in history:
-                prompt += f"<|im_start|>user\n{user_turn}<|im_end|>\n<|im_start|>assistant\n{assistant_turn}<|im_end|>\n"
-            prompt += f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
+        prompt = f"<|im_start|>system\n{system}<|im_end|>\n"
+        for (user_turn, assistant_turn) in history:
+            prompt += f"<|im_start|>user\n{user_turn}<|im_end|>\n<|im_start|>assistant\n{assistant_turn}<|im_end|>\n"
+        prompt += f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
 
         streamer = TextIteratorStreamer(pipe.tokenizer, timeout=240.0, skip_prompt=True, skip_special_tokens=True)
         generation_kwargs = dict(