Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
 # Use an official Python runtime as a parent image
-FROM python:3.8
+FROM python:3.9
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
@@ -17,6 +17,9 @@ WORKDIR $HOME/app
 # Clone your repository or add your code to the container
 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
+
 # Install Python dependencies
 RUN pip install -q einops pytorch_lightning gradio omegaconf xformers==0.0.20 transformers lpips opencv-python
 
@@ -47,4 +50,4 @@ RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co
 # EXPOSE 80
 
 # Define the command to run your application
-CMD ["python", "gradio_diffbir.py", "--ckpt", "$HOME/app/models/general_full_v1.ckpt", "--config", "$HOME/app/DiffBIR/configs/model/cldm.yaml", "--reload_swinir", "--swinir_ckpt", "$HOME/app/models/general_swinir_v1.ckpt"]
+CMD ["python", "gradio_diffbir.py", "--ckpt", "$HOME/app/models/general_full_v1.ckpt", "--config", "$HOME/app/configs/model/cldm.yaml", "--reload_swinir", "--swinir_ckpt", "$HOME/app/models/general_swinir_v1.ckpt"]