Update app.py
Browse files
app.py
CHANGED
@@ -146,7 +146,7 @@ def respond(message, history):
|
|
146 |
stream = client_mixtral.text_generation(messages, max_new_tokens=2000, do_sample=True, stream=True, details=True, return_full_text=False)
|
147 |
output = ""
|
148 |
for response in stream:
|
149 |
-
if not response.token.text == "
|
150 |
output += response.token.text
|
151 |
yield output
|
152 |
elif json_data["name"] == "image_generation":
|
|
|
146 |
stream = client_mixtral.text_generation(messages, max_new_tokens=2000, do_sample=True, stream=True, details=True, return_full_text=False)
|
147 |
output = ""
|
148 |
for response in stream:
|
149 |
+
if not response.token.text == "<|im_end|>":
|
150 |
output += response.token.text
|
151 |
yield output
|
152 |
elif json_data["name"] == "image_generation":
|