Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|