add cache env

Dockerfile

@@ -47,6 +47,13 @@ RUN mkdir -p pretrained && \
 # Add volume mount points for input and output files
 VOLUME ["/app/input", "/app/output"]
 
+# Create a directory for caches and make it writable
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+
+# Set environment variables to point to this new directory
+ENV NUMBA_CACHE_DIR=/app/cache/numba_cache
+ENV MPLCONFIGDIR=/app/cache/matplotlib_cache
+
 CMD ["python", "webui.py", "--work_dir", "pretrained", "--addr", "0.0.0.0", "--port", "7860"]
 
 # # Set entrypoint to run inference