fffiloni commited on
Commit
c85dadc
1 Parent(s): 6b4167a

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -32,11 +32,11 @@ RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co
32
  # Install dependencies
33
  RUN pip install --no-cache-dir -r environment.txt
34
 
35
- #COPY app.py .
36
 
37
  # Set the environment variable to specify the GPU device
38
  ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
39
  ENV CUDA_VISIBLE_DEVICES=0
40
 
41
  # Run your app.py script
42
- CMD ["python", "DragNUWA_demo.py"]
 
32
  # Install dependencies
33
  RUN pip install --no-cache-dir -r environment.txt
34
 
35
+ COPY app.py .
36
 
37
  # Set the environment variable to specify the GPU device
38
  ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
39
  ENV CUDA_VISIBLE_DEVICES=0
40
 
41
  # Run your app.py script
42
+ CMD ["python", "app.py"]