Update Dockerfile

Dockerfile

@@ -1,5 +1,20 @@
 FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
 ENV DEBIAN_FRONTEND=noninteractive
+
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:${PATH}
+WORKDIR ${HOME}/app
+
+COPY ./requirements.txt /code/requirements.txt
+COPY ./pre-requirements.txt /code/pre-requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/pre-requirements.txt
+RUN pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu117
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+
 RUN apt-get update && \
     apt-get upgrade -y && \
     apt-get install -y --no-install-recommends \
@@ -29,12 +44,6 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-RUN useradd -m -u 1000 user
-USER user
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:${PATH}
-WORKDIR ${HOME}/app
-
 RUN curl https://pyenv.run | bash
 ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
 ENV PYTHON_VERSION=3.10.9
@@ -56,4 +65,5 @@ ENV PYTHONPATH=${HOME}/app \
     GRADIO_SERVER_NAME=0.0.0.0 \
     GRADIO_THEME=huggingface \
     SYSTEM=spaces
+    
 CMD ["python", "app.py"]