Update app.py

app.py

@@ -99,9 +99,9 @@ def chat_llama3_8b(message: str,
     for text in streamer:
         # Remove thinking tags to prevent Gradio display issues
         if "<think>" in text:
-            text = text.replace("[think]", "").strip()
+            text = text.replace("<think>", "[think]").strip()
         if "</think>" in text:
-            text = text.replace("[/think]", "").strip()
+            text = text.replace("</think>", "[/think]").strip()
         outputs.append(text)
         print("".join(outputs))
         yield "".join(outputs)