feat: HF cache in dino class

models/dino.py

@@ -3,6 +3,13 @@ from transformers import GroundingDinoProcessor, GroundingDinoForObjectDetection
 
 import cv2
 
+import os
+
+HF_CACHE = "/tmp/hf_cache"
+os.makedirs(HF_CACHE, exist_ok=True)
+os.environ["TRANSFORMERS_CACHE"] = HF_CACHE
+
+
 
 class DinoWrapper:
     """

run_fastapi.py

@@ -15,10 +15,6 @@ 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))