Upload 6 files

README.md

@@ -4,7 +4,7 @@ emoji: 👁
 colorFrom: blue
 colorTo: pink
 sdk: gradio
-sdk_version: 5.17.0
+sdk_version: 5.20.0
 app_file: app.py
 pinned: true
 license: openrail

app.py

@@ -22,7 +22,6 @@ _REPO_ID = 'deepghs/anime_sites_indices'
 hf_fs = get_hf_fs()
 hf_client = get_hf_client()
 
-# _DEFAULT_MODEL_NAME = 'SwinV2_v3_iqdb_10_46796044_8GB'
 _DEFAULT_MODEL_NAME = 'SwinV2_v3_danbooru_8005009_4GB'
 _ALL_MODEL_NAMES = [
     os.path.dirname(os.path.relpath(path, _REPO_ID))
@@ -103,13 +102,13 @@ def search(model_name: str, img_input, str_input: str, n_neighbours: int):
     if str_input == "":
         embeddings = siglip.get_siglip_image_embedding(
             img_input,
-            model_name="smilingwolf/siglip_swinv2_base_2025_02_13_22h42m03s",
+            model_name="smilingwolf/siglip_swinv2_base_2025_02_22_18h56m54s",
             fmt="embeddings",
         )
     else:
         embeddings = siglip.get_siglip_text_embedding(
             str_input,
-            model_name="smilingwolf/siglip_swinv2_base_2025_02_13_22h42m03s",
+            model_name="smilingwolf/siglip_swinv2_base_2025_02_22_18h56m54s",
             fmt="embeddings",
         )
 

requirements.txt

@@ -2,6 +2,6 @@ pillow>=9.0.0
 faiss-cpu
 dghs-imgutils
 onnxruntime
-gradio==5.17.0
 cheesechaser>=0.1.6
-dghs-realutils
+dghs-realutils>=0.1.0
+gradio