Add experimental model

app.py

@@ -15,6 +15,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"
 CONV_MODEL_DSV3_REPO = "SmilingWolf/wd-convnext-tagger-v3"
@@ -29,6 +31,9 @@ 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"
+
 # Files to download from the repos
 MODEL_FILENAME = "model.onnx"
 LABEL_FILENAME = "selected_tags.csv"
@@ -62,7 +67,6 @@ def parse_args() -> argparse.Namespace:
     parser.add_argument("--score-slider-step", type=float, default=0.05)
     parser.add_argument("--score-general-threshold", type=float, default=0.35)
     parser.add_argument("--score-character-threshold", type=float, default=0.85)
-    parser.add_argument("--share", action="store_true")
     return parser.parse_args()
 
 
@@ -102,10 +106,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
 
@@ -229,11 +235,14 @@ 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,
+        # ---
+        EVA02_LARGE_MODEL_IS_DSV1_REPO,
     ]
 
     with gr.Blocks(title=TITLE) as demo: