SmilingWolf commited on
Commit
8246815
1 Parent(s): e610dd6

Add EVA02-Large model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,13 +15,12 @@ Demo for the WaifuDiffusion tagger models
15
  Example image by [ほし☆☆☆](https://www.pixiv.net/en/users/43565085)
16
  """
17
 
18
- HF_TOKEN = os.environ["HF_TOKEN"]
19
-
20
  # Dataset v3 series of models:
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"
@@ -231,6 +230,7 @@ def main():
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,
 
15
  Example image by [ほし☆☆☆](https://www.pixiv.net/en/users/43565085)
16
  """
17
 
 
 
18
  # Dataset v3 series of models:
19
  SWINV2_MODEL_DSV3_REPO = "SmilingWolf/wd-swinv2-tagger-v3"
20
  CONV_MODEL_DSV3_REPO = "SmilingWolf/wd-convnext-tagger-v3"
21
  VIT_MODEL_DSV3_REPO = "SmilingWolf/wd-vit-tagger-v3"
22
  VIT_LARGE_MODEL_DSV3_REPO = "SmilingWolf/wd-vit-large-tagger-v3"
23
+ EVA02_LARGE_MODEL_DSV3_REPO = "SmilingWolf/wd-eva02-large-tagger-v3"
24
 
25
  # Dataset v2 series of models:
26
  MOAT_MODEL_DSV2_REPO = "SmilingWolf/wd-v1-4-moat-tagger-v2"
 
230
  CONV_MODEL_DSV3_REPO,
231
  VIT_MODEL_DSV3_REPO,
232
  VIT_LARGE_MODEL_DSV3_REPO,
233
+ EVA02_LARGE_MODEL_DSV3_REPO,
234
  MOAT_MODEL_DSV2_REPO,
235
  SWIN_MODEL_DSV2_REPO,
236
  CONV_MODEL_DSV2_REPO,