Spaces:

mobinln
/

chatbot

Sleeping

mobinln commited on Feb 2

Commit

059b292

verified ·

1 Parent(s): ba9a48d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,8 +46,7 @@ def respond(
         delta = message['choices'][0]['delta']
         if 'content' in delta:
             response += delta['content']
-            formatted_response = re.sub(r'<think>(.*?)</think>', r'*\1*', response)
-            print(f"*- {response}")
             yield formatted_response

         delta = message['choices'][0]['delta']
         if 'content' in delta:
             response += delta['content']
+            formatted_response = re.sub(r"<think>\s*(.*?)\s*</think>", r"*\1*", response, flags=re.DOTALL)
             yield formatted_response