Update app.py

app.py

@@ -22,6 +22,8 @@ Demo for the WaifuDiffusion tagger models
 Example image by [ほし☆☆☆](https://www.pixiv.net/en/users/43565085)
 """
 
+HF_TOKEN = os.environ.get("HF_TOKEN", "") 
+
 
 # Dataset v3 series of models:
 SWINV2_MODEL_DSV3_REPO = "SmilingWolf/wd-swinv2-tagger-v3"
@@ -37,6 +39,12 @@ CONV_MODEL_DSV2_REPO = "SmilingWolf/wd-v1-4-convnext-tagger-v2"
 CONV2_MODEL_DSV2_REPO = "SmilingWolf/wd-v1-4-convnextv2-tagger-v2"
 VIT_MODEL_DSV2_REPO = "SmilingWolf/wd-v1-4-vit-tagger-v2"
 
+# IdolSankaku series of models:
+EVA02_LARGE_MODEL_IS_DSV1_REPO = "deepghs/idolsankaku-eva02-large-tagger-v1"
+SWINV2_MODEL_IS_DSV1_REPO = "deepghs/idolsankaku-swinv2-tagger-v1"
+
+
+
 # Files to download from the repos
 MODEL_FILENAME = "model.onnx"
 LABEL_FILENAME = "selected_tags.csv"
@@ -110,10 +118,12 @@ class Predictor:
         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,
         )
         return csv_path, model_path
 
@@ -223,7 +233,7 @@ class Predictor:
             " ".join(sorted_general_strings)
         )
         sorted_general_strings = (
-            ", ".join(sorted_general_strings).replace("(", "\(").replace(")", "\)")
+            ", ".join(sorted_general_strings).replace("(", r"\(").replace(")", r"\)")
         )
         # sorted_general_strings = sorted_general_strings.map(
         # lambda x: x.replace("_", " ") if x not in kaomojis else x
@@ -243,11 +253,15 @@ def main():
         VIT_MODEL_DSV3_REPO,
         VIT_LARGE_MODEL_DSV3_REPO,
         EVA02_LARGE_MODEL_DSV3_REPO,
+        # ---
         MOAT_MODEL_DSV2_REPO,
         SWIN_MODEL_DSV2_REPO,
         CONV_MODEL_DSV2_REPO,
         CONV2_MODEL_DSV2_REPO,
         VIT_MODEL_DSV2_REPO,
+        # ---
+        SWINV2_MODEL_IS_DSV1_REPO,
+        EVA02_LARGE_MODEL_IS_DSV1_REPO,
     ]
 
     with gr.Blocks(title=TITLE) as demo: