Update Dockerfile

Dockerfile

@@ -4,6 +4,24 @@ FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
 
+RUN useradd -m -u 1000 user
+
+USER user
+
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH \
+    PYTHONPATH=$HOME/app \
+	PYTHONUNBUFFERED=1 \
+	GRADIO_ALLOW_FLAGGING=never \
+	GRADIO_NUM_PORTS=1 \
+	GRADIO_SERVER_NAME=0.0.0.0 \
+	GRADIO_THEME=huggingface \
+    GRADIO_SHARE=False \
+	SYSTEM=spaces
+
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
@@ -21,9 +39,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.8 1
 RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
 
-# Create a working directory
-WORKDIR /app
-
 # Clone the repository (adjust the URL if needed)
 RUN git clone --recursive https://github.com/jnjaby/KEEP.git .
 
@@ -44,8 +59,9 @@ RUN pip install dlib ffmpeg-python
 # Initialize and update Git submodules (if not handled by the clone command)
 RUN git submodule init && git submodule update
 
-# Expose any required ports (if your app uses any)
-# EXPOSE 8000
+# Set the environment variable to specify the GPU device
+ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
+ENV CUDA_VISIBLE_DEVICES=0
 
 # Command to run your application
 CMD ["python", "app.py"]