fffiloni commited on
Commit
a4d092c
1 Parent(s): 8c69e8b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -45,5 +45,8 @@ RUN find $HOME/app
45
  ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
46
  ENV CUDA_VISIBLE_DEVICES=0
47
 
 
 
 
48
  # Run your app.py script
49
- CMD ["python", "/home/user/app/universal-image-restoration/config/daclip-sde/app.py"]
 
45
  ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
46
  ENV CUDA_VISIBLE_DEVICES=0
47
 
48
+ # Set the working directory to the user's home directory
49
+ WORKDIR $HOME/app/universal-image-restoration/config/daclip-sde/
50
+
51
  # Run your app.py script
52
+ CMD ["python", "app.py"]