KingNish commited on
Commit
6a8eb37
1 Parent(s): d382aa7

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 "Phi" in model:
33
  if output.endswith("<|assistant|>"):
34
- output = output(-13)
35
  else:
36
  if not response.token.text == "</s>":
37
  output += response.token.text
 
31
  for response in stream:
32
  if "Phi" in model:
33
  if output.endswith("<|assistant|>"):
34
+ output = output[:-13]
35
  else:
36
  if not response.token.text == "</s>":
37
  output += response.token.text