Update Dockerfile

Dockerfile

@@ -13,6 +13,12 @@ RUN apt-get update && apt-get install -y \
 # Set the working directory in the container
 WORKDIR /app
 
+# Create a new directory for Torch cache
+RUN mkdir -p /app/.cache/torch
+
+# Set environment variable to use the new cache location
+ENV TORCH_HOME=/app/.cache/torch
+
 # Copy the current directory contents into the container at /app
 COPY . /app