Update app.py

app.py

@@ -86,11 +86,15 @@ def respond(
     for message in res:
         #print(message)
         token = message.choices[0].delta.content
-        if token in ['<think>', '</think>']:
-            token = token.replace('<', '\\<').replace('>', '\\>')
+        #if token in ['<think>', '</think>']:
+        #    token = token.replace('<', '\\<').replace('>', '\\>')
         #print(type(token))
         response += token
+        if '<think>' in response:
+            response = response.replace('<think>', '\\<think\\>')
 
+        if '</think>' in response:
+            response = response.replace('</think>', '\\</think\\>')
         #print(response)
         yield response