Update app.py

app.py

@@ -88,7 +88,7 @@ def generate(text, past_key_values):
         prompt_format.format(system_message=system_prompt, prompt=text), **cur_generate_kwargs
     )[0]['generated_text']
     print(response)
-    return response.split('<|im_start|>assistant')[1]
+    return response.split('<|im_start|>assistant\n')[1]
     # return response[-1]['content']