Update app.py

app.py

@@ -82,8 +82,9 @@ except Exception as e:
     processor = None
 
 # SAM Wrapper
-@spaces.GPU
+
 class CustomSAMWrapper:
+    @spaces.GPU
     def __init__(self, model_path: str, device: str = DEVICE):
         # try:            
         self.device = torch.device(device)
@@ -95,7 +96,7 @@ 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],