feat: HF cache

run_fastapi.py

@@ -15,6 +15,11 @@ from models.dino import DinoWrapper
 
 from huggingface_hub import hf_hub_download
 
+# HF Cache
+HF_CACHE_DIR = os.path.join("/tmp", "hf_cache")
+os.environ["TRANSFORMERS_CACHE"] = HF_CACHE_DIR
+os.makedirs(HF_CACHE_DIR, exist_ok=True)
+
 # Check write permissions
 print("WRITE to ./weights:", os.access("weights", os.W_OK))
 print("WRITE to /tmp:", os.access("/tmp", os.W_OK))