pedrocas15 commited on
Commit
71f9478
·
verified ·
1 Parent(s): 8505b09

Update app.py

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