SmilingWolf commited on
Commit
e610dd6
1 Parent(s): 038b1de

Add ViT-Large model

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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,