SmilingWolf
commited on
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,
|