fffiloni commited on
Commit
9a6b64c
1 Parent(s): 555754c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -34,11 +34,11 @@ RUN pip install --no-cache-dir -r requirements.txt
34
  COPY pretrained/ $HOME/app/universal-image-restoration/config/daclip-sde/options/
35
 
36
  # Set the working directory to the app.py file's location
37
- WORKDIR $HOME/app/universal-image-restoration/config/daclip-sde/
38
 
39
  # Set the environment variable to specify the GPU device
40
  ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
41
  ENV CUDA_VISIBLE_DEVICES=0
42
 
43
  # Run your app.py script
44
- CMD ["python", "app.py"]
 
34
  COPY pretrained/ $HOME/app/universal-image-restoration/config/daclip-sde/options/
35
 
36
  # Set the working directory to the app.py file's location
37
+ WORKDIR $HOME/app/universal-image-restoration/
38
 
39
  # Set the environment variable to specify the GPU device
40
  ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
41
  ENV CUDA_VISIBLE_DEVICES=0
42
 
43
  # Run your app.py script
44
+ CMD ["python", "/config/daclip-sde/app.py"]