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

Update app.py

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