a9 commited on
Commit
56ab8f7
·
verified ·
1 Parent(s): 918ef3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -10
app.py CHANGED
@@ -37,21 +37,11 @@ SEG_MODELS = {
37
  "filename": "sapiens2_0.4b_seg.safetensors",
38
  "config": os.path.join(CONFIGS_DIR, "sapiens2_0.4b_seg_shutterstock_goliath-1024x768.py"),
39
  },
40
- "0.8B": {
41
- "repo": "facebook/sapiens2-seg-0.8b",
42
- "filename": "sapiens2_0.8b_seg.safetensors",
43
- "config": os.path.join(CONFIGS_DIR, "sapiens2_0.8b_seg_shutterstock_goliath-1024x768.py"),
44
- },
45
  "1B": {
46
  "repo": "facebook/sapiens2-seg-1b",
47
  "filename": "sapiens2_1b_seg.safetensors",
48
  "config": os.path.join(CONFIGS_DIR, "sapiens2_1b_seg_shutterstock_goliath-1024x768.py"),
49
  },
50
- "5B": {
51
- "repo": "facebook/sapiens2-seg-5b",
52
- "filename": "sapiens2_5b_seg.safetensors",
53
- "config": os.path.join(CONFIGS_DIR, "sapiens2_5b_seg_shutterstock_goliath-1024x768.py"),
54
- },
55
  }
56
  DEFAULT_SIZE = "1B"
57
 
 
37
  "filename": "sapiens2_0.4b_seg.safetensors",
38
  "config": os.path.join(CONFIGS_DIR, "sapiens2_0.4b_seg_shutterstock_goliath-1024x768.py"),
39
  },
 
 
 
 
 
40
  "1B": {
41
  "repo": "facebook/sapiens2-seg-1b",
42
  "filename": "sapiens2_1b_seg.safetensors",
43
  "config": os.path.join(CONFIGS_DIR, "sapiens2_1b_seg_shutterstock_goliath-1024x768.py"),
44
  },
 
 
 
 
 
45
  }
46
  DEFAULT_SIZE = "1B"
47