Update Dockerfile

Dockerfile

@@ -1,15 +1,17 @@
-# Use your existing image from Docker Hub
 FROM shahrukhsiddiq/diffusion-api:latest
 
 # Set working directory
 WORKDIR /app
 
-# Expose the port that your FastAPI runs on
-EXPOSE 8000
+# Ensure writable cache directory
+RUN mkdir -p /data && chmod -R 777 /data
 
-# Optional environment vars (improves caching on HF)
+# Set cache environment variables
 ENV TRANSFORMERS_CACHE=/data
 ENV HF_HOME=/data
 
-# Ensure correct entrypoint for FastAPI
+# Expose FastAPI port
+EXPOSE 8000
+
+# Launch FastAPI
 CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "8000"]