Spaces:

mobinln
/

chatbot

Sleeping

mobinln commited on Feb 2

Commit

d21f374

verified ·

1 Parent(s): c109793

fix bug in making thinking tokens italic

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,7 +45,7 @@ def respond(
     for message in completion:
         delta = message['choices'][0]['delta']
         if 'content' in delta:
-            # response += delta['content']
             formatted_response = re.sub(r'<think>(.*?)</think>', r'*\1*', response)
             yield formatted_response

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