Update Dockerfile

Dockerfile

@@ -23,15 +23,15 @@ RUN useradd -m -u 1000 user
 USER user
 
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
+    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 \
+    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
+    SYSTEM=spaces
 
 # Set CUDA_HOME environment variable
 ENV CUDA_HOME=/usr/local/cuda-11.8
@@ -42,34 +42,33 @@ ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
 ENV CUDA_VISIBLE_DEVICES=0
 
-# Clone the repository (adjust the URL if needed)
-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
 
-USER root
-# Install basicsr
-RUN python basicsr/setup.py develop
-
-USER user
+# Clone the repository (adjust the URL if needed)
+RUN git clone --recursive https://github.com/jnjaby/KEEP.git .
 
+# Copy the app.py script and requirements file into the container
+COPY --chown=user:user app.py .
+COPY --chown=user:user requirements_HF.txt .
 
-# Copy the app.py script into the container
-COPY app.py .
-COPY requirements_HF.txt .
+# Install Python dependencies
+RUN pip install --upgrade pip && \
+    pip install -r requirements_HF.txt && \
+    pip install gradio && \
+    pip install "numpy<1.25,>=1.18" && \
+    pip install cupy-cuda11x && \
+    pip install ffmpeg-python dlib
 
-# 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 root
+# Install basicsr
+RUN cd basicsr && \
+    if [ ! -f "basicsr/VERSION" ]; then \
+        echo "1.3.2" > basicsr/VERSION; \
+    fi && \
+    python setup.py develop
 
 USER user
-# Install additional Python packages
-RUN pip install ffmpeg-python
-RUN pip install dlib
 
 # Command to run your application
-CMD ["python", "app.py"]
+CMD ["python", "app.py"]