demo-ml-v3

Sleeping

spuun commited on Sep 19

Commit

a48be64

verified ·

1 Parent(s): 9f1b786

fix: thresh

Files changed (1) hide show

app.py CHANGED Viewed

@@ -87,7 +87,7 @@ def interface(classifier):
         # Use the normalized mask for visualization
         mask_viz = (normalized_mask * 255).astype(np.uint8)
-        binary_for_overlay = normalized_mask > 0.5
         overlay = image_np.copy()
         overlay[~binary_for_overlay] = (overlay[~binary_for_overlay] * 0.3).astype(np.uint8)

         # Use the normalized mask for visualization
         mask_viz = (normalized_mask * 255).astype(np.uint8)
+        binary_for_overlay = normalized_mask > 0.25
         overlay = image_np.copy()
         overlay[~binary_for_overlay] = (overlay[~binary_for_overlay] * 0.3).astype(np.uint8)