Hugging Face's logo

Spaces:
fffiloni
/
diffBIR
Paused

App Files Community
4
fffiloni commited on
Commit
9caeb0f
·
1 Parent(s): e2b5293

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -50,4 +50,4 @@ RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co
50
  # EXPOSE 80
51
 
52
  # Define the command to run your application
53
- CMD ["python", "gradio_diffbir.py", "--ckpt", "/models/general_full_v1.ckpt", "--config", "/DiffBIR/configs/model/cldm.yaml", "--reload_swinir", "--swinir_ckpt", "/models/general_swinir_v1.ckpt"]
 
50
  # EXPOSE 80
51
 
52
  # Define the command to run your application
53
+ CMD ["python", "gradio_diffbir.py", "--ckpt", "/home/user/app/models/general_full_v1.ckpt", "--config", "/home/user/app/configs/model/cldm.yaml", "--reload_swinir", "--swinir_ckpt", "/home/user/app/models/general_swinir_v1.ckpt"]