bleysg commited on
Commit
10eabc4
·
1 Parent(s): ac40c02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -53,8 +53,8 @@ def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetit
53
  for subtoken in tokens:
54
  subtoken = "".join(subtoken)
55
  # Remove "Response\n" if it's at the beginning of the assistant's output
56
- if subtoken.startswith("Response\n"):
57
- subtoken = subtoken[len("Response\n"):]
58
  answer = subtoken
59
  history[-1][1] += answer
60
  # stream the response
 
53
  for subtoken in tokens:
54
  subtoken = "".join(subtoken)
55
  # Remove "Response\n" if it's at the beginning of the assistant's output
56
+ if subtoken.startswith("Response"):
57
+ subtoken = subtoken[len("Response"):]
58
  answer = subtoken
59
  history[-1][1] += answer
60
  # stream the response