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,
|