Fix cache directory permissions and OMP_NUM_THREADS warnings - Set HF cache env vars before imports - Set MPLCONFIGDIR for matplotlib - Fix OMP_NUM_THREADS in Dockerfile

Dockerfile

@@ -48,6 +48,12 @@ ENV BASE_URL=${SPACE_HOST}
 ENV PORT=7860
 ENV DATA_DIR=/data
 ENV OMP_NUM_THREADS=1
+ENV HF_HOME=/tmp/hf_cache
+ENV TRANSFORMERS_CACHE=/tmp/hf_cache
+ENV HUGGINGFACE_HUB_CACHE=/tmp/hf_cache
+ENV HF_HUB_CACHE=/tmp/hf_cache
+ENV XDG_CACHE_HOME=/tmp/hf_cache
+ENV MPLCONFIGDIR=/tmp/matplotlib_config
 
 # Health check
 HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \

app/colorize_model.py

@@ -9,6 +9,15 @@ import logging
 import os
 from typing import Tuple
 
+# Ensure cache directory is set before any HF imports
+# (main.py should have set these, but ensure they're set here too)
+cache_dir = os.environ.get("HF_HOME", "/tmp/hf_cache")
+os.environ["HF_HOME"] = cache_dir
+os.environ["TRANSFORMERS_CACHE"] = cache_dir
+os.environ["HUGGINGFACE_HUB_CACHE"] = cache_dir
+os.environ["HF_HUB_CACHE"] = cache_dir
+os.environ["XDG_CACHE_HOME"] = cache_dir
+
 import torch
 from PIL import Image
 from fastai.vision.all import *
@@ -20,15 +29,17 @@ logger = logging.getLogger(__name__)
 
 
 def _ensure_cache_dir() -> str:
-    cache_dir = os.environ.get("HF_HOME") or "/tmp/hf_cache"
+    cache_dir = os.environ.get("HF_HOME", "/tmp/hf_cache")
     try:
         os.makedirs(cache_dir, exist_ok=True)
     except Exception as exc:
         logger.warning("Could not create cache directory %s: %s", cache_dir, exc)
+    # Ensure all cache env vars point to this directory
     os.environ["HF_HOME"] = cache_dir
     os.environ["TRANSFORMERS_CACHE"] = cache_dir
     os.environ["HUGGINGFACE_HUB_CACHE"] = cache_dir
     os.environ["HF_HUB_CACHE"] = cache_dir
+    os.environ["XDG_CACHE_HOME"] = cache_dir
     return cache_dir
 
 

app/main.py

@@ -3,8 +3,17 @@ FastAPI application for image colorization using ColorizeNet model
 with Firebase App Check integration
 """
 import os
+# Set environment variables BEFORE any imports to ensure they're used
 # Set OMP_NUM_THREADS before any torch imports to avoid libgomp warnings
-os.environ.setdefault("OMP_NUM_THREADS", "1")
+os.environ["OMP_NUM_THREADS"] = "1"
+# Set HF cache directories to writable /tmp location BEFORE any HF imports
+os.environ["HF_HOME"] = "/tmp/hf_cache"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+os.environ["HF_HUB_CACHE"] = "/tmp/hf_cache"
+os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/hf_cache"
+os.environ["XDG_CACHE_HOME"] = "/tmp/hf_cache"
+# Set matplotlib config directory to writable location
+os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib_config"
 
 import uuid
 import logging
@@ -24,13 +33,9 @@ import io
 from app.colorize_model import ColorizeModel
 from app.config import settings
 
-# Ensure HF caches and temporary directories use /tmp (writable in Spaces)
-os.environ.setdefault("HF_HOME", "/tmp/hf_cache")
-os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/hf_cache")
-os.environ.setdefault("HF_HUB_CACHE", "/tmp/hf_cache")
-os.environ.setdefault("HUGGINGFACE_HUB_CACHE", "/tmp/hf_cache")
-os.environ.setdefault("XDG_CACHE_HOME", "/tmp/hf_cache")
+# Create writable directories (env vars already set above)
 Path("/tmp/hf_cache").mkdir(parents=True, exist_ok=True)
+Path("/tmp/matplotlib_config").mkdir(parents=True, exist_ok=True)
 Path("/tmp/colorize_uploads").mkdir(parents=True, exist_ok=True)
 Path("/tmp/colorize_results").mkdir(parents=True, exist_ok=True)