pedrocas15 commited on
Commit
f579840
·
verified ·
1 Parent(s): 5da7f16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -54,7 +54,8 @@ def respond(
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):
 
54
  prompt = "<s>"
55
 
56
  for m in history:
57
+ prompt += f"{user_name}: {m[0].strip()}\n{ai_name}: {m[1].strip()}\n"
58
+ prompt += f"{ai_name}:"
59
 
60
  response = ""
61
  for tok in generate(prompt, use_rpc=use_rpc, max_tokens=max_tokens):