Update app.py

app.py

@@ -1,7 +1,6 @@
 import json
 import os
 from collections import defaultdict
-from functools import lru_cache
 from typing import List, Dict
 
 import faiss
@@ -13,6 +12,7 @@ from cheesechaser.datapool import YandeWebpDataPool, ZerochanWebpDataPool, Gelbo
 from hfutils.operate import get_hf_fs, get_hf_client
 from hfutils.utils import TemporaryDirectory
 from imgutils.tagging import wd14
+from imgutils.utils import ts_lru_cache
 
 from pools import quick_webp_pool
 
@@ -73,7 +73,7 @@ def _get_from_raw_ids(ids: List[str]) -> Dict[str, Image.Image]:
     return _retval
 
 
-@lru_cache(maxsize=3)
+@ts_lru_cache(maxsize=3)
 def _get_index_info(repo_id: str, model_name: str):
     image_ids = np.load(hf_client.hf_hub_download(
         repo_id=repo_id,