Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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("
|
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 |
|