Update app.py

app.py

@@ -237,14 +237,14 @@ def filter_mask_by_maxsize(masks, maximum_pixels):
     return renumbered_masks, removed_count
 
 
-def rec_min_size(masks):
+def rec_min_size(masks, q=25):
     ids = np.unique(masks)
     ids = ids[ids > 0]
-    num_cells = len(ids)
-    if num_cells <= 0:
+    if len(ids) == 0:
         return 0
-    mean_cell_size = np.count_nonzero(masks) / num_cells
-    return int(round(mean_cell_size))
+    sizes = np.array([np.count_nonzero(masks == cid) for cid in ids])
+    return int(round(np.percentile(sizes, q)))
+
 
 
 @spaces.GPU