Update app.py

app.py

@@ -48,6 +48,7 @@ def predict(message, history, system_prompt, temperature, max_tokens):
         temperature=0.5,
         max_new_tokens=DEFAULT_MAX_NEW_TOKENS,
         use_cache=True,
+        stop_token="<|im_end|>"
     )
     t = Thread(target=model.generate, kwargs=generate_kwargs)
     t.start()