Update Dockerfile

Dockerfile

@@ -8,9 +8,9 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-# Create directory for cache
-RUN mkdir -p /tmp/.cache/huggingface && \
-    chmod -R 777 /tmp/.cache
+# Create writable cache directories
+RUN mkdir -p /tmp/transformers_cache /tmp/hf_home /tmp/cache /tmp/diffusers_cache && \
+    chmod -R 777 /tmp
 
 # Copy requirements and install dependencies
 COPY requirements.txt .
@@ -19,6 +19,11 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy application code
 COPY app.py .
 
+# Set environment variables
+ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
+ENV HF_HOME=/tmp/hf_home
+ENV XDG_CACHE_HOME=/tmp/cache
+
 # Expose the port
 EXPOSE 7860