Update Dockerfile

Dockerfile

@@ -4,13 +4,20 @@ ENV DEBIAN_FRONTEND=noninteractive
 
 RUN pip3 install --no-cache-dir huggingface_hub huggingface_hub[cli]
 
-WORKDIR /code
+RUN useradd -m -u 1000 user
 
-COPY ./startup.sh /code/startup.sh
+USER user
 
-RUN chmod +x /code/startup.sh && cat /code/startup.sh
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
 
-COPY . .
+WORKDIR $HOME/code
+
+COPY --chown=user ./startup.sh $HOME/code/startup.sh
+
+RUN chmod +x $HOME/code/startup.sh && cat $HOME/code/startup.sh
+
+COPY --chown=user . $HOME/code
 
 EXPOSE 7860