TetAzivi commited on
Commit
b9ba61e
·
verified ·
1 Parent(s): e1bc466

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -35,6 +35,9 @@ VIT_MODEL_DSV2_REPO = "SmilingWolf/wd-v1-4-vit-tagger-v2"
35
  EVA02_LARGE_MODEL_IS_DSV1_REPO = "deepghs/idolsankaku-eva02-large-tagger-v1"
36
  SWINV2_MODEL_IS_DSV1_REPO = "deepghs/idolsankaku-swinv2-tagger-v1"
37
 
 
 
 
38
  # Files to download from the repos
39
  MODEL_FILENAME = "model.onnx"
40
  LABEL_FILENAME = "selected_tags.csv"
@@ -245,6 +248,8 @@ def main():
245
  # ---
246
  SWINV2_MODEL_IS_DSV1_REPO,
247
  EVA02_LARGE_MODEL_IS_DSV1_REPO,
 
 
248
  ]
249
 
250
  with gr.Blocks(title=TITLE) as demo:
 
35
  EVA02_LARGE_MODEL_IS_DSV1_REPO = "deepghs/idolsankaku-eva02-large-tagger-v1"
36
  SWINV2_MODEL_IS_DSV1_REPO = "deepghs/idolsankaku-swinv2-tagger-v1"
37
 
38
+ # test model:
39
+ VGG19_TEGGER_V3_REPO = "deepghs/wd-vgg19-tagger-v3"
40
+
41
  # Files to download from the repos
42
  MODEL_FILENAME = "model.onnx"
43
  LABEL_FILENAME = "selected_tags.csv"
 
248
  # ---
249
  SWINV2_MODEL_IS_DSV1_REPO,
250
  EVA02_LARGE_MODEL_IS_DSV1_REPO,
251
+ # ---
252
+ VGG19_TEGGER_V3_REPO,
253
  ]
254
 
255
  with gr.Blocks(title=TITLE) as demo: