geshang commited on
Commit
4667000
·
verified ·
1 Parent(s): e23fcaa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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}")