Update cache paths to use /tmp

Dockerfile

@@ -32,9 +32,9 @@ RUN mkdir -p /data/uploads /data/results
 # Note: Python code will create /data/hf_cache with proper permissions
 # Make Firebase credentials writing non-fatal (use || true to continue on error)
 RUN echo '#!/bin/sh' > /entrypoint.sh && \
-    echo 'mkdir -p /data/uploads /data/results || true' >> /entrypoint.sh && \
+    echo 'mkdir -p /tmp/colorize_uploads /tmp/colorize_results /tmp/hf_cache || true' >> /entrypoint.sh && \
     echo 'if [ -n "$FIREBASE_CREDENTIALS" ]; then' >> /entrypoint.sh && \
-    echo '  printf "%s" "$FIREBASE_CREDENTIALS" > /data/firebase-adminsdk.json 2>/dev/null || true' >> /entrypoint.sh && \
+    echo '  printf "%s" "$FIREBASE_CREDENTIALS" > /tmp/firebase-adminsdk.json 2>/dev/null || true' >> /entrypoint.sh && \
     echo 'fi' >> /entrypoint.sh && \
     echo 'exec "$@"' >> /entrypoint.sh && \
     chmod +x /entrypoint.sh

app/colorize_model.py

@@ -96,10 +96,14 @@ class ColorizeModel:
 
     def _load_caption_model(self) -> None:
         logger.info("Loading BLIP captioning model for prompt generation...")
-        self.caption_processor = BlipProcessor.from_pretrained(self.CAPTION_MODEL)
+        self.caption_processor = BlipProcessor.from_pretrained(
+            self.CAPTION_MODEL,
+            cache_dir=self.cache_dir
+        )
         self.caption_model = BlipForConditionalGeneration.from_pretrained(
             self.CAPTION_MODEL,
             torch_dtype=self.dtype if self.device.type == "cuda" else torch.float32,
+            cache_dir=self.cache_dir
         ).to(self.device)
 
     def caption_image(self, image: Image.Image) -> str:

app/main.py

@@ -21,6 +21,16 @@ 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")
+Path("/tmp/hf_cache").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)
+
 # Configure logging
 logging.basicConfig(
     level=logging.INFO,
@@ -45,7 +55,7 @@ app.add_middleware(
 )
 
 # Initialize Firebase Admin SDK
-firebase_cred_path = os.getenv("FIREBASE_CREDENTIALS_PATH", "colorize-662df-firebase-adminsdk-fbsvc-e080668793.json")
+firebase_cred_path = os.getenv("FIREBASE_CREDENTIALS_PATH", "/tmp/firebase-adminsdk.json")
 if os.path.exists(firebase_cred_path):
     try:
         cred = credentials.Certificate(firebase_cred_path)