Update app.py

app.py

@@ -142,6 +142,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
     buffer = ""
     for new_text in streamer:
         buffer += new_text
+        buffer = buffer.replace("<|im_end|>", "")
         time.sleep(0.01)
         yield buffer
 
@@ -209,6 +210,7 @@ def generate_video(model_name: str, text: str, video_path: str,
     buffer = ""
     for new_text in streamer:
         buffer += new_text
+        buffer = buffer.replace("<|im_end|>", "")
         time.sleep(0.01)
         yield buffer