Spaces:
Sleeping
Sleeping
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,
|