Spaces:
Running
Running
daquanzhou
commited on
Commit
·
6e26bf4
1
Parent(s):
2ffeae4
use gpu
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -47,4 +47,4 @@ WORKDIR $HOME/app
|
|
47 |
COPY --chown=user . $HOME/app
|
48 |
|
49 |
# Checkpoints
|
50 |
-
CMD ["python", "main.py", "--listen", "0.0.0.0", "--
|
|
|
47 |
COPY --chown=user . $HOME/app
|
48 |
|
49 |
# Checkpoints
|
50 |
+
CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "7860", "--output-directory", "${USE_PERSISTENT_DATA:+/data/}"]
|