Update Dockerfile

Dockerfile

@@ -1,6 +1,6 @@
 FROM nvidia/cuda:12.6.0-runtime-ubuntu22.04
 
-RUN useradd -m -u 1001 user
+RUN useradd -m -u 1000 user
 # System depsFROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
 
 RUN apt update && apt upgrade -y
@@ -17,6 +17,9 @@ RUN apt-get install -y build-essential
 RUN apt-get install -y ninja-build
 RUN apt-get install -y python3.10-venv
 
+ENV TRANSFORMERS_CACHE=/home/user/.cache/huggingface
+RUN mkdir -p /home/user/.cache/huggingface
+
 # Create virtualenv
 RUN python3 -m venv /opt/venv
 ENV PATH="/opt/venv/bin:$PATH"