Spaces:
Paused
Paused
Update modules/masking_module.py
Browse files
modules/masking_module.py
CHANGED
@@ -23,7 +23,7 @@ models_dict = {
|
|
23 |
SAM_CHECKPOINT = "/home/user/app/sam2_hiera_large.pt"
|
24 |
SAM_CONFIG = "sam2_hiera_l.yaml"
|
25 |
|
26 |
-
@spaces.GPU(duration=
|
27 |
def load_sam_image_model(
|
28 |
device: torch.device,
|
29 |
config: str = SAM_CONFIG,
|
|
|
23 |
SAM_CHECKPOINT = "/home/user/app/sam2_hiera_large.pt"
|
24 |
SAM_CONFIG = "sam2_hiera_l.yaml"
|
25 |
|
26 |
+
@spaces.GPU(duration=20)
|
27 |
def load_sam_image_model(
|
28 |
device: torch.device,
|
29 |
config: str = SAM_CONFIG,
|