Denliner commited on
Commit
3861450
1 Parent(s): 8a6e711

update the app and remove the security issue

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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,