Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -304,6 +304,7 @@ def run_pipeline(image: PILImage.Image, prompt: str):
|
|
304 |
masks_np=[mask_np],
|
305 |
alpha=0.6
|
306 |
)
|
|
|
307 |
return output_text, visualized_img
|
308 |
except Exception as e:
|
309 |
print(f"Pipeline error: {e}")
|
|
|
304 |
masks_np=[mask_np],
|
305 |
alpha=0.6
|
306 |
)
|
307 |
+
output_text = re.findall(r'<think>(.*?)</think>', output_text, re.DOTALL)
|
308 |
return output_text, visualized_img
|
309 |
except Exception as e:
|
310 |
print(f"Pipeline error: {e}")
|