push app

app.py

@@ -121,9 +121,11 @@ else:
         masks_tensor = results[0].masks.data
         masks = masks_tensor.cpu().numpy()
         if masks.ndim == 3 and masks.shape[0] > 0:
-            # Combine masks (logical OR via max)
+             # Combine masks (logical OR via max)
             combined_mask = np.max(masks, axis=0)
             combined_mask_img = Image.fromarray((combined_mask * 255).astype(np.uint8))
+            # Resize the mask to ensure it matches the base image size
+            combined_mask_img = combined_mask_img.resize(img.size, Image.NEAREST)
             # Create a red overlay with transparency
             overlay = Image.new("RGBA", img.size, (255, 0, 0, 100))
             base = img.convert("RGBA")