Update app.py

app.py

@@ -85,16 +85,16 @@ except Exception as e:
 class CustomSAMWrapper:
     @spaces.GPU
     def __init__(self, model_path: str, device: str = DEVICE):
-        try:            
-            self.device = torch.device(device)
-            sam_model = build_sam2("configs/sam2.1/sam2.1_hiera_l.yaml", model_path, self.device)
-            sam_model = sam_model.to(self.device)
-            self.predictor = SAM2ImagePredictor(sam_model)
-            self.last_mask = None
-            print(f"SAM model loaded on {device}")
-        except Exception as e:
-            print(f"Error loading SAM model: {e}")
-            self.predictor = None
+        # try:            
+        self.device = torch.device(device)
+        sam_model = build_sam2("configs/sam2.1/sam2.1_hiera_l.yaml", model_path, self.device)
+        sam_model = sam_model.to(self.device)
+        self.predictor = SAM2ImagePredictor(sam_model)
+        self.last_mask = None
+        print(f"SAM model loaded on {device}")
+        # except Exception as e:
+        #     print(f"Error loading SAM model: {e}")
+        #     self.predictor = None
 
     def predict(self, image: PILImage.Image, 
                points: List[Tuple[int, int]],