Update Dockerfile

Dockerfile

@@ -1,14 +1,16 @@
-# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
-# you will also find guides on how best to write your Dockerfile
-
 # Base image
 FROM python:3.9
 
 # Set the working directory in the container
 WORKDIR /code
 
-# Set an environment variable for Matplotlib cache
+# Set environment variables for writable directories
 ENV MPLCONFIGDIR=/tmp/matplotlib_cache
+ENV XDG_CACHE_HOME=/tmp/cache
+
+# Create the directories and ensure permissions
+RUN mkdir -p /tmp/cache \
+    && chmod -R 777 /tmp
 
 # Install libGL for OpenCV
 RUN apt-get update && apt-get install -y \
@@ -28,5 +30,3 @@ COPY . .
 
 # Command to run the application
 CMD ["python", "code/demo.py"]
-
-