Update Dockerfile

Dockerfile

@@ -12,7 +12,8 @@ RUN mkdir -p /cache && \
 ENV TRANSFORMERS_CACHE=/cache \
     HF_HOME=/cache \
     XDG_CACHE_HOME=/cache \
-    PYTHONPATH=/code
+    PYTHONPATH=/code \
+    TORCH_HOME=/cache
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -45,7 +46,8 @@ COPY ./models /code/models
 
 # Set proper permissions for the application
 RUN chown -R 1000:1000 /code && \
-    chmod -R 755 /code
+    chmod -R 755 /code && \
+    chmod -R 777 /cache
 
 EXPOSE 7860