Update Dockerfile

Dockerfile

@@ -2,11 +2,20 @@ ARG TORCH_VERSION=2.3.0
 ARG CUDA_VERSION=12.1
 FROM pytorch/pytorch:${TORCH_VERSION}-cuda${CUDA_VERSION}-cudnn8-runtime
 
-WORKDIR /opt/NekoImageGallery
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+
+# Switch to the "user" user
+USER user
+
+# Set home to the user's home directory
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME/opt/NekoImageGallery
 
 COPY requirements.txt .
-RUN mkdir /root/.cache/huggingface/hub
-RUN chmod -R 777 /root/.cache/huggingface/hub
+
 RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -r requirements.txt
 
 RUN mkdir -p /opt/models && \