Update Dockerfile

Dockerfile

@@ -22,7 +22,7 @@ WORKDIR $HOME/app
 RUN git clone -b dev https://github.com/camenduru/DiffBIR $HOME/app
 
 # Install PyTorch and torchvision with a specific CUDA toolkit version
-RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
+#RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
 
 # Install Python dependencies
 RUN pip install -q einops pytorch_lightning gradio omegaconf xformers==0.0.20 transformers lpips opencv-python