docker

Dockerfile

@@ -26,7 +26,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY --chown=user:user app.py .
 
 # Create cache directory with correct permissions
-RUN mkdir -p /home/user/.cache /tmp/transformers /tmp/huggingface /tmp/torch && chown -R user:user /home/user/.cache
+RUN mkdir -p /home/user/.cache /tmp/transformers /tmp/huggingface /tmp/torch && chown -R user:user /home/user/.cache /tmp/transformers /tmp/huggingface /tmp/torch
 
 # Switch to non-root user
 USER user

app.py

@@ -107,8 +107,29 @@ class ZeroShotFoodClassifier:
 
         logger.info(f"🚀 Loading CLIP model: {self.model_name}")
 
-        # Centralizovan cache u /tmp da izbjegnemo prekomjerno trajno skladištenje
-        cache_dir = os.environ.get("HF_CACHE_DIR", os.environ.get("TRANSFORMERS_CACHE", "/tmp/transformers"))
+        # Centralizovan cache u /tmp; prefer HF_HOME, zatim TRANSFORMERS_CACHE
+        hf_home = os.environ.get("HF_HOME")
+        cache_dir = hf_home if hf_home else os.environ.get("TRANSFORMERS_CACHE", "/tmp/transformers")
+
+        # Ensure cache directories exist and are writable; clean stale locks
+        try:
+            os.makedirs(cache_dir, exist_ok=True)
+            # Transformers also uses a models subdir sometimes; ensure base is writable
+            for root_dir in {cache_dir, "/tmp/huggingface", "/tmp/torch"}:
+                try:
+                    os.makedirs(root_dir, exist_ok=True)
+                except Exception:
+                    pass
+            # Remove stale lock files that can block downloads
+            for dirpath, dirnames, filenames in os.walk(cache_dir):
+                for filename in filenames:
+                    if filename.endswith(".lock") or filename.endswith("-partial"):  # defensive
+                        try:
+                            os.remove(os.path.join(dirpath, filename))
+                        except Exception:
+                            pass
+        except Exception as e:
+            logger.warning(f"⚠️ Cache setup warning: {e}")
 
         load_kwargs: Dict[str, Any] = {"cache_dir": cache_dir}
         if self.device in ("cuda", "mps"):
@@ -125,6 +146,11 @@ class ZeroShotFoodClassifier:
                 if torch.cuda.is_available():
                     torch.cuda.empty_cache()
                 self.model_name = FALLBACK_MODEL_NAME
+                # On fallback, also retry ensuring cache writability and cleaning locks
+                try:
+                    os.makedirs(cache_dir, exist_ok=True)
+                except Exception:
+                    pass
                 self.processor = CLIPProcessor.from_pretrained(self.model_name, cache_dir=cache_dir)
                 fallback_kwargs = load_kwargs.copy()
                 self.model = CLIPModel.from_pretrained(self.model_name, **fallback_kwargs).to(self.device)