Update app.py

app.py

@@ -19,6 +19,7 @@ def chat_fn(message, history):
             temperature=0.7
         )
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    response = response.replace("<|assistant|>", "").replace("<|user|>", "").strip()
     return response
 
 demo = gr.ChatInterface(