Update Dockerfile

Dockerfile

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