Update Dockerfile

Dockerfile

@@ -24,9 +24,8 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
     wget \
     && rm -rf /var/lib/apt/lists/*
 
-RUN useradd -m -u 1000 user
-USER user
-
+# RUN useradd -m -u 1000 user
+# USER user
 
 # RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
 RUN pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
@@ -35,6 +34,8 @@ RUN pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url h
 
 ARG USER_NAME=berf
 ARG GROUP_NAME=berfs
+ARG UID=1000
+ARG GID=1000
 # Change user to non-root user
 RUN groupadd -g ${GID} ${GROUP_NAME} \
     && useradd -ms /bin/sh -u ${UID} -g ${GID} ${USER_NAME}