Update Dockerfile

Dockerfile

@@ -59,8 +59,8 @@ RUN pip install gradio
 # Install specific Cython version known to work with cupy
 RUN pip install 'Cython<3.0'
 
-# Install cupy before basicsr setup
-RUN pip install cupy==10.4.0
+# Install cupy using the pre-built wheel for CUDA 11.x
+RUN pip install --no-cache-dir cupy-cuda11x==10.4.0
 
 USER root
 # Configure git to trust the directory