Update app.py

app.py

@@ -304,7 +304,9 @@ def run_pipeline(image: PILImage.Image, prompt: str):
             masks_np=[mask_np],
             alpha=0.6
         )
-        output_text = re.findall(r'<think>(.*?)</think>', output_text, re.DOTALL)
+        match = re.search(r'(<think>.*?</think>)', output_text, re.DOTALL)
+        if match:
+            output_text = match.group(1)
         return output_text, visualized_img
     except Exception as e:
         print(f"Pipeline error: {e}")