Update app.py

app.py

@@ -352,6 +352,14 @@ def _build_device_map(model: Any) -> tuple[dict[str, Any], str]:
     return device_map, same_device_modules[0]
 
 
+def _clean_reasoning_trace(text: str) -> str:
+    for token in ("<|im_start|>", "<|im_end|>", "<|endoftext|>"):
+        text = text.replace(token, "")
+    end_tag = "</think>"
+    if end_tag in text:
+        text = text.split(end_tag, 1)[0] + end_tag
+    return text
+
 def ensure_runtime() -> tuple[Any, Any]:
     inferencer = _runtime_cache["inferencer"]
     pil_img2rgb = _runtime_cache["pil_img2rgb"]
@@ -515,7 +523,7 @@ def edit_image(
             result_text += chunk
         else:
             last_image = chunk
-        yield last_image, result_text
+        yield last_image, _clean_reasoning_trace(result_text)
 
 
 def update_edit_thinking_visibility(show: bool):