Hugging Face's logo

Spaces:
microhan
/
efficientSAM_mask
Sleeping

App Files Community
microhan commited on
Commit
bdf9307
·
verified ·
1 Parent(s): 565c48b

Update utils/tools_gradio.py

Browse files
Files changed (1) hide show
  1. utils/tools_gradio.py +2 -1
utils/tools_gradio.py CHANGED
@@ -89,7 +89,8 @@ def fast_process(
89
  overlay_contour = Image.fromarray((contour_mask * 255).astype(np.uint8), "RGBA")
90
  image.paste(overlay_contour, (0, 0), overlay_contour)
91
 
92
- return image
 
93
 
94
 
95
  # CPU post process
 
89
  overlay_contour = Image.fromarray((contour_mask * 255).astype(np.uint8), "RGBA")
90
  image.paste(overlay_contour, (0, 0), overlay_contour)
91
 
92
+ # return image
93
+ return inner_mask
94
 
95
 
96
  # CPU post process