Update app.py

app.py

@@ -13,7 +13,7 @@ DESCRIPTION = """
 Demo for the WaifuDiffusion tagger models
 """
 
-HF_TOKEN = os.environ.get("HF_TOKEN", "")
+HF_TOKEN1 = os.environ.get("HF_TOKEN1", "")
 
 # Dataset v3 series of models:
 SWINV2_MODEL_DSV3_REPO = "SmilingWolf/wd-swinv2-tagger-v3"
@@ -56,8 +56,8 @@ class Predictor:
         self.last_loaded_repo = None
 
     def download_model(self, model_repo):
-        csv_path = huggingface_hub.hf_hub_download(model_repo, LABEL_FILENAME, use_auth_token=HF_TOKEN)
-        model_path = huggingface_hub.hf_hub_download(model_repo, MODEL_FILENAME, use_auth_token=HF_TOKEN)
+        csv_path = huggingface_hub.hf_hub_download(model_repo, LABEL_FILENAME, use_auth_token=HF_TOKEN1)
+        model_path = huggingface_hub.hf_hub_download(model_repo, MODEL_FILENAME, use_auth_token=HF_TOKEN1)
         return csv_path, model_path
 
     def load_model(self, model_repo):
@@ -144,6 +144,12 @@ def main():
                        "2024",
                        "2023",
                        "2025",
+                       "genderswap",
+                       "genderswap_(otm)",
+                       "genderswap_(otf)",
+                       "genderswap_(mtf)",
+                       "genderswap_(ftm)",
+                       "respirator",
                        "head-mounted_display",
                        "2022",
                        "muscular_female",