Spaces:

yonigozlan
/

edgetam

Running on Zero

yonigozlan HF Staff commited on Sep 26

Commit

6d0e912

1 Parent(s): e2bed97

update edgetam checkpoint

Files changed (1) hide show

app.py CHANGED Viewed

@@ -171,7 +171,7 @@ def _model_repo_from_key(key: str) -> str:
         "small": "facebook/sam2.1-hiera-small",
         "base_plus": "facebook/sam2.1-hiera-base-plus",
         "large": "facebook/sam2.1-hiera-large",
-        "EdgeTAM": "../EdgeTAM-hf",
     }
     return mapping.get(key, mapping["base_plus"])

         "small": "facebook/sam2.1-hiera-small",
         "base_plus": "facebook/sam2.1-hiera-base-plus",
         "large": "facebook/sam2.1-hiera-large",
+        "EdgeTAM": "yonigozlan/EdgeTAM-hf",
     }
     return mapping.get(key, mapping["base_plus"])