Update Dockerfile

Dockerfile

@@ -3,12 +3,13 @@ FROM shahrukhsiddiq/diffusion-api:latest
 # Set working directory
 WORKDIR /app
 
-# Ensure writable cache directory
-RUN mkdir -p /data && chmod -R 777 /data
+# Create writable cache folder
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+
+# Redirect Hugging Face and Transformers cache
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV HF_HOME=/app/cache
 
-# Set cache environment variables
-ENV TRANSFORMERS_CACHE=/data
-ENV HF_HOME=/data
 
 # Expose FastAPI port
 EXPOSE 8000