Spaces:
Running
Running
Commit
·
e610dd6
1
Parent(s):
038b1de
Add ViT-Large model
Browse files
app.py
CHANGED
|
@@ -21,6 +21,7 @@ HF_TOKEN = os.environ["HF_TOKEN"]
|
|
| 21 |
SWINV2_MODEL_DSV3_REPO = "SmilingWolf/wd-swinv2-tagger-v3"
|
| 22 |
CONV_MODEL_DSV3_REPO = "SmilingWolf/wd-convnext-tagger-v3"
|
| 23 |
VIT_MODEL_DSV3_REPO = "SmilingWolf/wd-vit-tagger-v3"
|
|
|
|
| 24 |
|
| 25 |
# Dataset v2 series of models:
|
| 26 |
MOAT_MODEL_DSV2_REPO = "SmilingWolf/wd-v1-4-moat-tagger-v2"
|
|
@@ -229,6 +230,7 @@ def main():
|
|
| 229 |
SWINV2_MODEL_DSV3_REPO,
|
| 230 |
CONV_MODEL_DSV3_REPO,
|
| 231 |
VIT_MODEL_DSV3_REPO,
|
|
|
|
| 232 |
MOAT_MODEL_DSV2_REPO,
|
| 233 |
SWIN_MODEL_DSV2_REPO,
|
| 234 |
CONV_MODEL_DSV2_REPO,
|
|
|
|
| 21 |
SWINV2_MODEL_DSV3_REPO = "SmilingWolf/wd-swinv2-tagger-v3"
|
| 22 |
CONV_MODEL_DSV3_REPO = "SmilingWolf/wd-convnext-tagger-v3"
|
| 23 |
VIT_MODEL_DSV3_REPO = "SmilingWolf/wd-vit-tagger-v3"
|
| 24 |
+
VIT_LARGE_MODEL_DSV3_REPO = "SmilingWolf/wd-vit-large-tagger-v3"
|
| 25 |
|
| 26 |
# Dataset v2 series of models:
|
| 27 |
MOAT_MODEL_DSV2_REPO = "SmilingWolf/wd-v1-4-moat-tagger-v2"
|
|
|
|
| 230 |
SWINV2_MODEL_DSV3_REPO,
|
| 231 |
CONV_MODEL_DSV3_REPO,
|
| 232 |
VIT_MODEL_DSV3_REPO,
|
| 233 |
+
VIT_LARGE_MODEL_DSV3_REPO,
|
| 234 |
MOAT_MODEL_DSV2_REPO,
|
| 235 |
SWIN_MODEL_DSV2_REPO,
|
| 236 |
CONV_MODEL_DSV2_REPO,
|