Update app.py

app.py

@@ -376,7 +376,7 @@ def generate_text(input_text, messages=None):
 
             if output.endswith("<|end_header_id|>\n\n"):
                 continue
-            html_out = output.replace("<|endoftext|>", "").replace("<|begin_of_text|>", "").strip().split("<|end_header_id|>\n\n")[-1].replace("**", "")
+            html_out = output.replace("<|endoftext|>", "").replace("<|begin_of_text|>", "").strip().split("<|end_header_id|>")[-1].replace("**", "")
             
             yield [messages[-1]] + [{"role": "assistant", "content": html_out}]
     return