Update app.py

app.py

@@ -82,7 +82,7 @@ def predict(input, chatbot, max_length, top_p, temperature, history):
 """
     optimized_prompt += resp
     chatbot = [(parse_text(input), parse_text(optimized_prompt))]
-    yield chatbot, history
+    # yield chatbot, history
 
     optimized_prompt += "\n\n====================Aggressive Optimization===================="
 
@@ -90,8 +90,6 @@ def predict(input, chatbot, max_length, top_p, temperature, history):
     responses = []
     num = 0
     for text in texts:
-        chatbot = [(parse_text(input), parse_text(optimized_prompt))]
-        yield chatbot, history
         num += 1
         seed = torch.seed()
         torch.manual_seed(seed)
@@ -107,8 +105,8 @@ def predict(input, chatbot, max_length, top_p, temperature, history):
         optimized_prompt += f"\n{num}. {resp}"
 
         chatbot = [(parse_text(input), parse_text(optimized_prompt))]
-        yield chatbot, history
-
+        # yield chatbot, history
+    return chatbot, history
 
 
 def reset_user_input():