Update Dockerfile

Dockerfile

@@ -8,9 +8,9 @@ WORKDIR /code
 ENV MPLCONFIGDIR=/tmp/matplotlib_cache
 ENV XDG_CACHE_HOME=/tmp/cache
 
-# Create and set permissions for the gdown cache directory
-RUN mkdir -p /.cache/gdown \
-    && chmod -R 777 /.cache
+# Create necessary directories and set permissions
+RUN mkdir -p /.cache/gdown /code/pretrained_models \
+    && chmod -R 777 /.cache /code/pretrained_models
 
 # Install libGL for OpenCV
 RUN apt-get update && apt-get install -y \