Spaces:

xu-song
/

self-chat

Sleeping

xu song commited on Jul 31

Commit

e4e3ccf

•

1 Parent(s): 2e49897

update

Files changed (1) hide show

app_util.py CHANGED Viewed

@@ -62,13 +62,16 @@ def generate_response(chatbot, history, user_input=None):
 def generate(chatbot, history):
     if history[-1]["role"] == "assistant":
-        return generate_response(chatbot, history)
     elif history[-1]["role"] in ["user", "system"]:
-        return generate_query(chatbot, history)
     else:
         gr.Warning("bug")
-        return
 def regenerate():

 def generate(chatbot, history):
+    streamer = None
     if history[-1]["role"] == "assistant":
+        streamer = generate_response(chatbot, history)
     elif history[-1]["role"] in ["user", "system"]:
+        streamer = generate_query(chatbot, history)
     else:
         gr.Warning("bug")
+    for out in streamer:
+        yield out
 def regenerate():