diff --git a/web_demo_mm.py b/web_demo_mm.py
index 753df09..896861e 100644
--- a/web_demo_mm.py
+++ b/web_demo_mm.py
@@ -164,7 +164,7 @@ def regenerate(_chatbot, task_history):
             _chatbot[-1] = (_chatbot[-1][0], None)
         else:
             _chatbot.append((chatbot_item[0], None))
-        return predict(_chatbot, task_history)
+        yield from predict(_chatbot, task_history)
 
     def add_text(history, task_history, text):
         task_text = text