search_image_by_image_or_text

Running

narugo1992 commited on Feb 17

Commit

05e8d01

verified ·

1 Parent(s): 32f7aca

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,8 @@ _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'
 _ALL_MODEL_NAMES = [
     os.path.dirname(os.path.relpath(path, _REPO_ID))
     for path in hf_fs.glob(f'{_REPO_ID}/*/knn.index')

 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))
     for path in hf_fs.glob(f'{_REPO_ID}/*/knn.index')