SmilingWolf commited on
Commit
eaba46a
1 Parent(s): db51843

Add another experimental model

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -33,6 +33,7 @@ VIT_MODEL_DSV2_REPO = "SmilingWolf/wd-v1-4-vit-tagger-v2"
33
 
34
  # IdolSankaku series of models:
35
  EVA02_LARGE_MODEL_IS_DSV1_REPO = "deepghs/idolsankaku-eva02-large-tagger-v1"
 
36
 
37
  # Files to download from the repos
38
  MODEL_FILENAME = "model.onnx"
@@ -242,6 +243,7 @@ def main():
242
  CONV2_MODEL_DSV2_REPO,
243
  VIT_MODEL_DSV2_REPO,
244
  # ---
 
245
  EVA02_LARGE_MODEL_IS_DSV1_REPO,
246
  ]
247
 
 
33
 
34
  # IdolSankaku series of models:
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"
 
243
  CONV2_MODEL_DSV2_REPO,
244
  VIT_MODEL_DSV2_REPO,
245
  # ---
246
+ SWINV2_MODEL_IS_DSV1_REPO,
247
  EVA02_LARGE_MODEL_IS_DSV1_REPO,
248
  ]
249