feat: use better model checkpoint
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -17,4 +17,4 @@ COPY --chown=user utils $HOME/app/utils
|
|
17 |
COPY --chown=user configs $HOME/app/configs
|
18 |
COPY --chown=user demo.py $HOME/app/demo.py
|
19 |
|
20 |
-
CMD ["python", "demo.py", "-d", "-1", "-c", "huggingface://gyrojeff/YuzuMarker.FontDetection/
|
|
|
17 |
COPY --chown=user configs $HOME/app/configs
|
18 |
COPY --chown=user demo.py $HOME/app/demo.py
|
19 |
|
20 |
+
CMD ["python", "demo.py", "-d", "-1", "-c", "huggingface://gyrojeff/YuzuMarker.FontDetection/name=4x-epoch=10-step=213444.ckpt", "-m", "resnet50", "-z", "512", "-p", "7860", "-a", "0.0.0.0"]
|