initial commit

modules/sam.py

@@ -54,11 +54,13 @@ class SamInference:
             'min_mask_region_area': int(params[5]),
         }
 
-        if self.model is None or self.mask_generator is None or self.tunable_params != tunable_params:
-            self.tunable_params = tunable_params
-            self.set_mask_generator()
-
-        masks = self.mask_generator.generate(image)
-        combined_image = create_mask_combined_images(image, masks)
-        gallery = create_mask_gallery(image, masks)
-        return [combined_image] + gallery
+        try:
+            if self.model is None or self.mask_generator is None or self.tunable_params != tunable_params:
+                self.tunable_params = tunable_params
+                self.set_mask_generator()
+            masks = self.mask_generator.generate(image)
+            combined_image = create_mask_combined_images(image, masks)
+            gallery = create_mask_gallery(image, masks)
+            return [combined_image] + gallery
+        except Exception as e:
+            print(e)