Update Dockerfile

Dockerfile

@@ -7,10 +7,11 @@ WORKDIR /code
 # Set environment variables for writable directories
 ENV MPLCONFIGDIR=/tmp/matplotlib_cache
 ENV XDG_CACHE_HOME=/tmp/cache
+ENV TORCH_HOME=/tmp/cache/torch
 
 # Create necessary directories and set permissions
-RUN mkdir -p /.cache/gdown /code/pretrained_models \
-    && chmod -R 777 /.cache /code/pretrained_models
+RUN mkdir -p /.cache/gdown /code/pretrained_models /tmp/cache/torch \
+    && chmod -R 777 /.cache /code/pretrained_models /tmp/cache
 
 # Install libGL for OpenCV
 RUN apt-get update && apt-get install -y \