fffiloni commited on
Commit
5602f66
·
1 Parent(s): 02b684c

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", "$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"]
 
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", "/configs/model/cldm.yaml", "--reload_swinir", "--swinir_ckpt", "/models/general_swinir_v1.ckpt"]