Nithish310 commited on
Commit
c2fc75b
·
verified ·
1 Parent(s): c3b2412

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -171,7 +171,7 @@ def respond(message, history):
171
  output = ""
172
  for response in stream:
173
  if not response.token.text == "hello":
174
- output += response.token.text
175
  yield output
176
  elif json_data["name"] == "image_generation":
177
  query = json_data["arguments"]["query"]
 
171
  output = ""
172
  for response in stream:
173
  if not response.token.text == "hello":
174
+ output += response.token.text.replace("<|im_end|>", "")
175
  yield output
176
  elif json_data["name"] == "image_generation":
177
  query = json_data["arguments"]["query"]