Spaces:
Runtime error
Runtime error
Commit
·
078f137
1
Parent(s):
53a263b
Update app.py
Browse files
app.py
CHANGED
@@ -17,14 +17,6 @@ import PIL.Image
|
|
17 |
from Utils import dbimutils
|
18 |
|
19 |
TITLE = "WaifuDiffusion v1.4 Tags"
|
20 |
-
DESCRIPTION = """
|
21 |
-
Demo for:
|
22 |
-
- [SmilingWolf/wd-v1-4-moat-tagger-v2](https://huggingface.co/SmilingWolf/wd-v1-4-moat-tagger-v2)
|
23 |
-
- [SmilingWolf/wd-v1-4-swinv2-tagger-v2](https://huggingface.co/SmilingWolf/wd-v1-4-convnext-tagger-v2)
|
24 |
-
- [SmilingWolf/wd-v1-4-convnext-tagger-v2](https://huggingface.co/SmilingWolf/wd-v1-4-convnext-tagger-v2)
|
25 |
-
- [SmilingWolf/wd-v1-4-convnextv2-tagger-v2](https://huggingface.co/SmilingWolf/wd-v1-4-convnextv2-tagger-v2)
|
26 |
-
- [SmilingWolf/wd-v1-4-vit-tagger-v2](https://huggingface.co/SmilingWolf/wd-v1-4-vit-tagger-v2)
|
27 |
-
"""
|
28 |
|
29 |
HF_TOKEN = os.environ["HF_TOKEN"]
|
30 |
MOAT_MODEL_REPO = "SmilingWolf/wd-v1-4-moat-tagger-v2"
|
@@ -218,7 +210,7 @@ def main():
|
|
218 |
|
219 |
args = parse_args()
|
220 |
|
221 |
-
change_model("
|
222 |
|
223 |
tag_names, rating_indexes, general_indexes, character_indexes = load_labels()
|
224 |
|
@@ -236,7 +228,7 @@ def main():
|
|
236 |
gr.Image(type="pil", label="Input"),
|
237 |
gr.Radio(
|
238 |
["MOAT", "SwinV2", "ConvNext", "ConvNextV2", "ViT"],
|
239 |
-
value="
|
240 |
label="Model",
|
241 |
),
|
242 |
gr.Slider(
|
@@ -264,7 +256,6 @@ def main():
|
|
264 |
],
|
265 |
examples=[],
|
266 |
title=TITLE,
|
267 |
-
description=DESCRIPTION,
|
268 |
allow_flagging="never",
|
269 |
).launch(
|
270 |
enable_queue=True,
|
|
|
17 |
from Utils import dbimutils
|
18 |
|
19 |
TITLE = "WaifuDiffusion v1.4 Tags"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
HF_TOKEN = os.environ["HF_TOKEN"]
|
22 |
MOAT_MODEL_REPO = "SmilingWolf/wd-v1-4-moat-tagger-v2"
|
|
|
210 |
|
211 |
args = parse_args()
|
212 |
|
213 |
+
change_model("MOAT")
|
214 |
|
215 |
tag_names, rating_indexes, general_indexes, character_indexes = load_labels()
|
216 |
|
|
|
228 |
gr.Image(type="pil", label="Input"),
|
229 |
gr.Radio(
|
230 |
["MOAT", "SwinV2", "ConvNext", "ConvNextV2", "ViT"],
|
231 |
+
value="MOAT",
|
232 |
label="Model",
|
233 |
),
|
234 |
gr.Slider(
|
|
|
256 |
],
|
257 |
examples=[],
|
258 |
title=TITLE,
|
|
|
259 |
allow_flagging="never",
|
260 |
).launch(
|
261 |
enable_queue=True,
|