Update Dockerfile

Dockerfile

@@ -10,8 +10,13 @@ COPY requirements.txt .
 RUN pip install --upgrade pip && pip install --no-cache-dir -r requirements.txt
 
 # Set Hugging Face cache for all model downloads (best practice for Spaces)
+# ENV HF_HOME="/workspace/.cache/huggingface"
+# RUN mkdir -p /workspace/.cache/huggingface
+ENV CACHE_DIR="/workspace/.cache"
 ENV HF_HOME="/workspace/.cache/huggingface"
-RUN mkdir -p /workspace/.cache/huggingface
+ENV TRANSFORMERS_CACHE="/workspace/.cache/huggingface"
+RUN mkdir -p /workspace/.cache/huggingface/hub && \
+    chmod -R 777 /workspace/.cache
 
 # Pre-download Stable Diffusion v1-5 weights into cache
 RUN python -c "import torch; from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='/workspace/.cache/huggingface', torch_dtype=torch.float32)"