pedrocas15 commited on
Commit
abcef1f
·
verified ·
1 Parent(s): ac2a08a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -54,9 +54,11 @@ def respond(
54
  prompt = "<s>"
55
 
56
  for m in history:
57
- prompt += f"{user_name}: {m[0]}\n{ai_name}: {m[1]}\n"
58
-
59
- for response in generate(prompt, use_rpc=use_rpc, max_tokens=max_tokens)
 
 
60
  yield response
61
 
62
 
 
54
  prompt = "<s>"
55
 
56
  for m in history:
57
+ prompt += f"{user_name}: {m[0]}\n{ai_name}: {m[1]}"
58
+
59
+ response = ""
60
+ for tok in generate(prompt, use_rpc=use_rpc, max_tokens=max_tokens)
61
+ respond += tok
62
  yield response
63
 
64