Spaces:
Running
Running
update the app and remove the security issue
Browse files
app.py
CHANGED
@@ -22,12 +22,13 @@ Demo for the WaifuDiffusion tagger models
|
|
22 |
Example image by [ほし☆☆☆](https://www.pixiv.net/en/users/43565085)
|
23 |
"""
|
24 |
|
25 |
-
HF_TOKEN = os.environ["HF_TOKEN"]
|
26 |
|
27 |
# Dataset v3 series of models:
|
28 |
SWINV2_MODEL_DSV3_REPO = "SmilingWolf/wd-swinv2-tagger-v3"
|
29 |
CONV_MODEL_DSV3_REPO = "SmilingWolf/wd-convnext-tagger-v3"
|
30 |
VIT_MODEL_DSV3_REPO = "SmilingWolf/wd-vit-tagger-v3"
|
|
|
|
|
31 |
|
32 |
# Dataset v2 series of models:
|
33 |
MOAT_MODEL_DSV2_REPO = "SmilingWolf/wd-v1-4-moat-tagger-v2"
|
@@ -109,12 +110,10 @@ class Predictor:
|
|
109 |
csv_path = huggingface_hub.hf_hub_download(
|
110 |
model_repo,
|
111 |
LABEL_FILENAME,
|
112 |
-
use_auth_token=HF_TOKEN,
|
113 |
)
|
114 |
model_path = huggingface_hub.hf_hub_download(
|
115 |
model_repo,
|
116 |
MODEL_FILENAME,
|
117 |
-
use_auth_token=HF_TOKEN,
|
118 |
)
|
119 |
return csv_path, model_path
|
120 |
|
@@ -242,6 +241,8 @@ def main():
|
|
242 |
SWINV2_MODEL_DSV3_REPO,
|
243 |
CONV_MODEL_DSV3_REPO,
|
244 |
VIT_MODEL_DSV3_REPO,
|
|
|
|
|
245 |
MOAT_MODEL_DSV2_REPO,
|
246 |
SWIN_MODEL_DSV2_REPO,
|
247 |
CONV_MODEL_DSV2_REPO,
|
|
|
22 |
Example image by [ほし☆☆☆](https://www.pixiv.net/en/users/43565085)
|
23 |
"""
|
24 |
|
|
|
25 |
|
26 |
# Dataset v3 series of models:
|
27 |
SWINV2_MODEL_DSV3_REPO = "SmilingWolf/wd-swinv2-tagger-v3"
|
28 |
CONV_MODEL_DSV3_REPO = "SmilingWolf/wd-convnext-tagger-v3"
|
29 |
VIT_MODEL_DSV3_REPO = "SmilingWolf/wd-vit-tagger-v3"
|
30 |
+
VIT_LARGE_MODEL_DSV3_REPO = "SmilingWolf/wd-vit-large-tagger-v3"
|
31 |
+
EVA02_LARGE_MODEL_DSV3_REPO = "SmilingWolf/wd-eva02-large-tagger-v3"
|
32 |
|
33 |
# Dataset v2 series of models:
|
34 |
MOAT_MODEL_DSV2_REPO = "SmilingWolf/wd-v1-4-moat-tagger-v2"
|
|
|
110 |
csv_path = huggingface_hub.hf_hub_download(
|
111 |
model_repo,
|
112 |
LABEL_FILENAME,
|
|
|
113 |
)
|
114 |
model_path = huggingface_hub.hf_hub_download(
|
115 |
model_repo,
|
116 |
MODEL_FILENAME,
|
|
|
117 |
)
|
118 |
return csv_path, model_path
|
119 |
|
|
|
241 |
SWINV2_MODEL_DSV3_REPO,
|
242 |
CONV_MODEL_DSV3_REPO,
|
243 |
VIT_MODEL_DSV3_REPO,
|
244 |
+
VIT_LARGE_MODEL_DSV3_REPO,
|
245 |
+
EVA02_LARGE_MODEL_DSV3_REPO,
|
246 |
MOAT_MODEL_DSV2_REPO,
|
247 |
SWIN_MODEL_DSV2_REPO,
|
248 |
CONV_MODEL_DSV2_REPO,
|