Update Dockerfile

Dockerfile

@@ -47,11 +47,13 @@ RUN git clone --recursive https://github.com/jnjaby/KEEP.git .
 
 # 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.txt
+RUN pip install -r requirements_HF.txt
 RUN pip install gradio
+RUN pip install cupy==10.4.0
 
 USER root
 # Install basicsr (assuming setup.py is in basicsr directory)
@@ -59,7 +61,8 @@ RUN python basicsr/setup.py develop
 
 USER user
 # Install additional Python packages
-RUN pip install dlib ffmpeg-python
+RUN pip install ffmpeg-python
+RUN pip install dlib
 
 # Set the environment variable to specify the GPU device
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID