4eJIoBek commited on
Commit
1f43411
1 Parent(s): 785a887

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -14,6 +14,6 @@ COPY stable_diffusion_engine.py demo.py demo_web.py /src/
14
  COPY data/ /src/data/
15
 
16
  # download models
17
- RUN python3 demo_web.py --num-inference-steps 1 --prompt "test" --output /tmp/test.jpg
18
 
19
- ENTRYPOINT ["python3", "demo_web.py"]
 
14
  COPY data/ /src/data/
15
 
16
  # download models
17
+ RUN python3 demo.py --num-inference-steps 1 --prompt "test" --output /tmp/test.jpg
18
 
19
+ ENTRYPOINT ["python3", "demo.py"]