KingNish commited on
Commit
32a93fe
1 Parent(s): 7388d5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def models(text, model="Mixtral 8x7B"):
31
  for response in stream:
32
  if not response.token.text == "</s>":
33
  output += response.token.text
34
- if output.endswith("<end_of_utterance>"):
35
  output = output[:-13]
36
  return output
37
 
 
31
  for response in stream:
32
  if not response.token.text == "</s>":
33
  output += response.token.text
34
+ if output.endswith("<|assistant|>"):
35
  output = output[:-13]
36
  return output
37