bleysg commited on
Commit
ac40c02
·
1 Parent(s): 929c3e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -52,6 +52,9 @@ def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetit
52
  tokens = re.findall(r'(.*?)(\s|$)', tokens)
53
  for subtoken in tokens:
54
  subtoken = "".join(subtoken)
 
 
 
55
  answer = subtoken
56
  history[-1][1] += answer
57
  # stream the response
 
52
  tokens = re.findall(r'(.*?)(\s|$)', tokens)
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