Hugging Face's logo

Spaces:
yonigozlan
/
Segment-Anything-2-video-tracking
Running on Zero

App Files Community
2
yonigozlan HF Staff commited on
Commit
ba25bef
·
1 Parent(s): 4df052a

update checkpoints

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -160,10 +160,10 @@ class AppState:
160
 
161
  def _model_repo_from_key(key: str) -> str:
162
  mapping = {
163
- "tiny": "yonigozlan/sam2.1_hiera_tiny_hf",
164
- "small": "yonigozlan/sam2.1_hiera_small_hf",
165
- "base_plus": "yonigozlan/sam2.1_hiera_base_plus_hf",
166
- "large": "yonigozlan/sam2.1_hiera_large_hf",
167
  }
168
  return mapping.get(key, mapping["base_plus"])
169
 
 
160
 
161
  def _model_repo_from_key(key: str) -> str:
162
  mapping = {
163
+ "tiny": "facebook/sam2.1-hiera-tiny",
164
+ "small": "facebook/sam2.1-hiera-small",
165
+ "base_plus": "facebook/sam2.1-hiera-base-plus",
166
+ "large": "facebook/sam2.1-hiera-large",
167
  }
168
  return mapping.get(key, mapping["base_plus"])
169