Update cache to save memory

crs_arena/arena.py

@@ -61,7 +61,7 @@ if not os.path.exists("data/models"):
     logger.info("Downloading models...")
     download_and_extract_models()
     # Cache some CRS fighters at startup
-    cache_fighters()
+    cache_fighters(n=6)
 if not os.path.exists("data/embed_items"):
     logger.info("Downloading item embeddings...")
     download_and_extract_item_embeddings()

crs_arena/utils.py

@@ -25,7 +25,9 @@ from src.model.crs_model import CRSModel
 HF_API = HfApi(token=st.secrets["hf_token"])
 
 
-@st.cache_resource(show_spinner="Loading CRS...")
+@st.cache_resource(
+    show_spinner="Loading CRS...", max_entries=6, ttl=timedelta(days=1)
+)
 def get_crs_model(model_name: str, model_config_file: str) -> CRSModel:
     """Returns a CRS model.