Geek7 commited on
Commit
7985018
·
verified ·
1 Parent(s): be0f93a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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":