Spaces:

microhan
/

efficientSAM_mask

Sleeping

microhan commited on Mar 1

Commit

bdf9307

•

1 Parent(s): 565c48b

Update utils/tools_gradio.py

Files changed (1) hide show

utils/tools_gradio.py CHANGED Viewed

@@ -89,7 +89,8 @@ def fast_process(
         overlay_contour = Image.fromarray((contour_mask * 255).astype(np.uint8), "RGBA")
         image.paste(overlay_contour, (0, 0), overlay_contour)
-    return image
 # CPU post process

         overlay_contour = Image.fromarray((contour_mask * 255).astype(np.uint8), "RGBA")
         image.paste(overlay_contour, (0, 0), overlay_contour)
+    # return image
+    return inner_mask
 # CPU post process