xu song commited on
Commit
2b9d822
·
1 Parent(s): e52ef2a
Files changed (1) hide show
  1. app_util.py +2 -2
app_util.py CHANGED
@@ -25,7 +25,7 @@ def generate_query(chatbot, history):
25
  yield None, chatbot, history
26
  else:
27
  chatbot.append(None)
28
- streamer = bot.generate_query(history[-1], stream=True)
29
  for query, query_tokens in streamer:
30
  chatbot[-1] = (query, None)
31
  yield query, chatbot, history
@@ -44,7 +44,7 @@ def generate_response(chatbot, history):
44
  gr.Warning('You should generate or type user-input first.')
45
  yield None, chatbot, history
46
  else:
47
- streamer = bot.generate_response(history[-1], stream=True)
48
  for response, response_tokens in streamer:
49
  chatbot[-1] = (query, response)
50
  yield response, chatbot, history
 
25
  yield None, chatbot, history
26
  else:
27
  chatbot.append(None)
28
+ streamer = bot.generate_query(history, stream=True)
29
  for query, query_tokens in streamer:
30
  chatbot[-1] = (query, None)
31
  yield query, chatbot, history
 
44
  gr.Warning('You should generate or type user-input first.')
45
  yield None, chatbot, history
46
  else:
47
+ streamer = bot.generate_response(history, stream=True)
48
  for response, response_tokens in streamer:
49
  chatbot[-1] = (query, response)
50
  yield response, chatbot, history