Spaces:

Tianfong
/

moeflow-companion

Running

jokester commited on Nov 24, 2024

Commit

75f6754

1 Parent(s): 200ca07

download detection models in Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -7,8 +7,9 @@ WORKDIR /app
 RUN mkdir -pv /app \
         && micromamba run -n mit-py311 python3 -mvenv --system-site-packages /app/venv
 RUN venv/bin/pip install -r /home/mambauser/bootstrap/requirements-moeflow.txt
-# commands above are cacheable
 # (must specify -n base to use installed curl)
 RUN micromamba run -n base curl -L https://github.com/moeflow-com/manga-image-translator/archive/58ada4f51cf54b2e784d38c50543f9be8a9f28e4.tar.gz  | tar xvz --strip-components=1
-RUN venv/bin/python docker_prepare.py --models ocr.48px,ocr.48px_ctc,ocr.32px,ocr.mocr
 CMD venv/bin/python gradio-multi.py

 RUN mkdir -pv /app \
         && micromamba run -n mit-py311 python3 -mvenv --system-site-packages /app/venv
 RUN venv/bin/pip install -r /home/mambauser/bootstrap/requirements-moeflow.txt
+# HF does not seem to have layer cache anyway
 # (must specify -n base to use installed curl)
 RUN micromamba run -n base curl -L https://github.com/moeflow-com/manga-image-translator/archive/58ada4f51cf54b2e784d38c50543f9be8a9f28e4.tar.gz  | tar xvz --strip-components=1
+# NOTE for unknown reason we need to download here. or there will be runtime file permission error
+RUN venv/bin/python docker_prepare.py --models ocr.48px,ocr.48px_ctc,ocr.32px,ocr.mocr,detector.default,detector.ctd,detector.craft,detector.none
 CMD venv/bin/python gradio-multi.py