Pectics commited on
Commit
89132db
·
1 Parent(s): 2d8a02a

Fix python generator error

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -63,7 +63,8 @@ def respond(
63
  for m in history:
64
  messages.append({"role": m["role"], "content": m["content"]})
65
  messages.append({"role": "user", "content": message})
66
- return infer(messages, max_tokens, temperature, top_p)
 
67
 
68
  app = gr.ChatInterface(
69
  respond,
 
63
  for m in history:
64
  messages.append({"role": m["role"], "content": m["content"]})
65
  messages.append({"role": "user", "content": message})
66
+ for response in infer(messages, max_tokens, temperature, top_p):
67
+ yield response
68
 
69
  app = gr.ChatInterface(
70
  respond,