fix: temporarily disable ZeroGPU
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ choice_mapping = {
|
|
| 20 |
"large": ["sam2_hiera_l.yaml", "assets/checkpoints/sam2_hiera_large.pt"],
|
| 21 |
}
|
| 22 |
|
| 23 |
-
@spaces.GPU
|
| 24 |
def predict(model_choice: str, annotations, image):
|
| 25 |
config_file, ckpt_path = choice_mapping[str(model_choice)]
|
| 26 |
sam2_model = build_sam2(config_file, ckpt_path, device="cuda")
|
|
|
|
| 20 |
"large": ["sam2_hiera_l.yaml", "assets/checkpoints/sam2_hiera_large.pt"],
|
| 21 |
}
|
| 22 |
|
| 23 |
+
# @spaces.GPU
|
| 24 |
def predict(model_choice: str, annotations, image):
|
| 25 |
config_file, ckpt_path = choice_mapping[str(model_choice)]
|
| 26 |
sam2_model = build_sam2(config_file, ckpt_path, device="cuda")
|