Fix caching by writing only in /tmp dir.

Dockerfile

@@ -28,19 +28,19 @@ WORKDIR /app
 # Install runtime dependency: libopenblas.so.0 is provided by libopenblas-base.
 RUN apt-get update && apt-get install -y build-essential cmake libopenblas-dev liblapack-dev libopenblas-dev liblapack-dev
 
-# Create the "model" directory with appropriate permissions
-RUN mkdir -p /app/model && chmod -R 777 /app/model
+RUN mkdir -p /tmp/model /tmp/huggingface /tmp/.cache /tmp/.gradio \
+    && chmod -R 777 /tmp/model /tmp/huggingface /tmp/.cache /tmp/.gradio
 
-COPY --from=builder /app/venv venv
+ENV HF_HOME=/tmp/huggingface
+ENV HUGGINGFACE_HUB_CACHE=/tmp/huggingface
+ENV XDG_CACHE_HOME=/tmp/.cache
+ENV GRADIO_CACHE_DIR=/tmp/.gradio
 
-COPY app.py .
-COPY models.py .
-COPY test_functions.py .
+COPY --from=builder /app/venv venv
 
+COPY app.py models.py test_functions.py ./
 COPY examples/ /app/examples/
-RUN chmod -R 777 /app/examples/
-
-COPY assets/ assets/
+COPY assets/ /app/assets/
 
 ENV VIRTUAL_ENV=/app/venv
 ENV PATH="$VIRTUAL_ENV/bin:$PATH"

app.py

@@ -10,9 +10,14 @@ from huggingface_hub import hf_hub_download
 import tempfile
 import requests
 
-# Ensure model directory exists
-os.makedirs("model", exist_ok=True)
-MODEL_PATH = "model/best_unet_model.pth"
+os.environ["HF_HOME"]             = "/tmp/huggingface"
+os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface"
+os.environ["XDG_CACHE_HOME"]      = "/tmp/.cache"
+os.environ["GRADIO_CACHE_DIR"]    = "/tmp/.gradio"
+
+MODEL_DIR = "/tmp/model"
+os.makedirs(MODEL_DIR, exist_ok=True)
+MODEL_PATH = os.path.join(MODEL_DIR, "best_unet_model.pth")
 
 # Download model if missing
 def download_model():
@@ -20,8 +25,8 @@ def download_model():
     path = hf_hub_download(
         repo_id="Robys01/face-aging",
         filename="best_unet_model.pth",
-        local_dir="model",
-        cache_dir="model"
+        local_dir="MODEL_DIR",
+        cache_dir=os.environ["HUGGINGFACE_HUB_CACHE"],
     )
     print(f"Model downloaded to {path}")