Update sam_module.py

sam_module.py

@@ -71,10 +71,10 @@ def create_sam_mask(self, image, bbox_coords, mode):
             for i in range(num_masks):
                 single_mask = outputs.pred_masks[:, :, i, :, :]
                 resized_mask = F.interpolate(
-                single_mask,
-                size=(image.height, image.width),
-                mode='bilinear',
-               align_corners=False
+                    single_mask,
+                    size=(image.height, image.width),
+                    mode='bilinear',
+                    align_corners=False
              ).squeeze()
     
             mask_np = resized_mask.sigmoid().cpu().numpy()
@@ -341,10 +341,10 @@ def create_sam_mask(self, image, bbox_coords, mode):
             for i in range(num_masks):
                 single_mask = outputs.pred_masks[:, :, i, :, :]
                 resized_mask = F.interpolate(
-                single_mask,
-                size=(image.height, image.width),
-                mode='bilinear',
-                align_corners=False
+                    single_mask,
+                    size=(image.height, image.width),
+                    mode='bilinear',
+                    align_corners=False
             ).squeeze()
     
             mask_np = resized_mask.sigmoid().cpu().numpy()