Update Dockerfile

Dockerfile

@@ -1,18 +1,25 @@
 # Use an official CUDA runtime as a parent image
 FROM nvidia/cuda:11.2.2-runtime-ubuntu20.04
 
-# Set the working directory in the container
-WORKDIR /usr/src/app
+# Copy the current directory contents into the container at /usr/src/app
+WORKDIR $HOME/app
 
 # Install Python
 RUN apt-get update && apt-get install -y python3.8 python3-pip
 
+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
+ 
 # Set the cache directories to a path with write access
 ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
 ENV MPLCONFIGDIR=/tmp/matplotlib
+ENV HF_HOME=/tmp/huggingface_cache
 
-# Copy the current directory contents into the container at /usr/src/app
-COPY . /usr/src/app
+COPY --chown=user . $HOME/app
 
 # Install any needed packages specified in requirements.txt
 RUN pip3 install --no-cache-dir -r requirements.txt