Update Dockerfile

Dockerfile

@@ -42,38 +42,12 @@ ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
 ENV CUDA_VISIBLE_DEVICES=0
 
-
-
-
-# Install system dependencies as root
-USER root
-
-# Set Python 3.8 as the default python and pip versions
-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
-
 # Clone the repository (adjust the URL if needed)
-RUN git clone --recursive https://github.com/jnjaby/KEEP.git .
-
-# Set Git ownership for the cloned repository
-USER root
-RUN git config --global --add safe.directory $HOME/app
-USER user
+RUN git clone --recursive https://github.com/jnjaby/KEEP.git $HOME/app
 
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
 
-# Copy the app.py script into the container
-COPY app.py .
-COPY requirements_HF.txt .
-
-# Install Python dependencies from requirements.txt
-RUN pip install --upgrade pip
-RUN pip install -r requirements_HF.txt
-RUN pip install gradio
-RUN pip install "numpy<1.25,>=1.18"
-RUN pip install cupy==10.4.0
-
 USER root
 # Install basicsr
 RUN if [ -d "basicsr" ]; then \
@@ -88,6 +62,20 @@ RUN if [ -d "basicsr" ]; then \
     fi
 
 USER user
+
+
+# Copy the app.py script into the container
+COPY app.py .
+COPY requirements_HF.txt .
+
+# Install Python dependencies from requirements.txt
+RUN pip install --upgrade pip
+RUN pip install -r requirements_HF.txt
+RUN pip install gradio
+RUN pip install "numpy<1.25,>=1.18"
+RUN pip install cupy-wheel
+
+USER user
 # Install additional Python packages
 RUN pip install ffmpeg-python
 RUN pip install dlib