Update Dockerfile

Dockerfile

@@ -2,11 +2,12 @@ FROM python:3.9
 
 WORKDIR /code
 
-# Create a directory for the cache
-RUN mkdir -p /code/cache
+# Create a cache directory and set permissions
+RUN mkdir -p /code/cache && \
+    chmod -R 777 /code/cache
 
-# Set environment variable to use this directory for the Transformers cache
-ENV TRANSFORMERS_CACHE=/code/cache
+# Set environment variable for the Transformers cache
+ENV HF_HOME=/code/cache
 # Install segment-anything from GitHub
 RUN pip install --no-cache-dir git+https://github.com/facebookresearch/segment-anything.git