Update app.py

app.py

@@ -84,10 +84,9 @@ except Exception as e:
 # SAM Wrapper
 
 class CustomSAMWrapper:
-    @spaces.GPU
     def __init__(self, model_path: str, device: str = DEVICE):
         # try:            
-        self.device = torch.device(device)
+        self.device = "cpu" #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)
@@ -96,7 +95,6 @@ class CustomSAMWrapper:
         # except Exception as e:
         #     print(f"Error loading SAM model: {e}")
         #     self.predictor = None
-    @spaces.GPU
     def predict(self, image: PILImage.Image, 
                points: List[Tuple[int, int]], 
                labels: List[int],