Hugging Face's logo

Spaces:
AkaneKurokawa
/
wd-tagger
Running

App Files Community
SmilingWolf commited on
Commit
f2540ce
•
1 Parent(s): 9ee88e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -215,8 +215,7 @@ def main():
215
 
216
  args = parse_args()
217
 
218
- swin_model = load_model(SWIN_MODEL_REPO, MODEL_FILENAME)
219
- loaded_models["SwinV2"] = swin_model
220
 
221
  tag_names, rating_indexes, general_indexes, character_indexes = load_labels()
222
 
 
215
 
216
  args = parse_args()
217
 
218
+ change_model("SwinV2")
 
219
 
220
  tag_names, rating_indexes, general_indexes, character_indexes = load_labels()
221