SmilingWolf
commited on
Commit
•
8246815
1
Parent(s):
e610dd6
Add EVA02-Large model
Browse files
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,
|