RefalMachine commited on
Commit
9a067cb
·
verified ·
1 Parent(s): 97f173f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -86,6 +86,8 @@ def respond(
86
  for message in res:
87
  #print(message)
88
  token = message.choices[0].delta.content
 
 
89
  print(type(token))
90
  response += token
91
 
 
86
  for message in res:
87
  #print(message)
88
  token = message.choices[0].delta.content
89
+ if token in ['<think>', '</think>']:
90
+ token = token.replace('<', '\\<').replace('>', '\\>')
91
  print(type(token))
92
  response += token
93