torch issue resolved and added cache dir

Dockerfile

@@ -4,6 +4,12 @@ FROM python:3.9-slim
 # Set working directory
 WORKDIR /app
 
+ENV HOME=/app
+
+RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
+ENV XDG_CACHE_HOME=/app/.cache
+
+
 # Install required system packages
 RUN apt-get update && apt-get install -y \
     libgl1-mesa-glx \