TechnoByte commited on
Commit
078f137
·
1 Parent(s): 53a263b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -11
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("ConvNextV2")
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="ConvNextV2",
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,