gyrojeff commited on
Commit
9e89388
1 Parent(s): 80f623e

feat: use better model checkpoint

Browse files
Files changed (1) hide show
  1. 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/commit=bc0f7fc-epoch=26-step=261954.ckpt", "-m", "resnet50", "-z", "512", "-p", "7860", "-a", "0.0.0.0"]
 
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"]