mask overlay corrected

opengeos · Oct 22, 2024 · 510c626 · 510c626
2 parents c682fea + 9d30aa0
commit 510c626
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/samgeo/text_sam.py b/samgeo/text_sam.py
@@ -342,8 +342,10 @@ def predict(
 
         if boxes.nelement() == 0:  # No "object" instances found
             print("No objects found in the image.")
-            mask_overlay = np.zeros_like(image_np[..., 0], dtype=dtype)  # Create an empty mask overlay
-
+            mask_overlay = np.zeros_like(
+                image_np[..., 0], dtype=dtype
+            )  # Create an empty mask overlay
+
         else:
             # Create an empty image to store the mask overlays
             mask_overlay = np.zeros_like(