bellingcat/whisperbox-transcribe
1
0
Fork 0
mirror of https://github.com/bellingcat/whisperbox-transcribe.git synced 2026-06-11 21:18:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat: allow multiple models to be installed in docker container

Browse Source
This commit is contained in:
Felix Spöttel
2023-02-28 22:05:09 +01:00
parent 099a0eb1d7
commit 20f558584a
6 changed files with 18 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docker/dev/worker.Dockerfile
View File

@@ -1,6 +1,6 @@
FROM python:3.10
ARG WHISPER_MODEL
ARG WHISPER_MODELS
WORKDIR /code
@@ -13,7 +13,7 @@ COPY --from=mwader/static-ffmpeg:5.1.2 /ffprobe /usr/local/bin/
COPY pyproject.toml .
RUN pip install --no-cache-dir --user .[worker,worker_dev]
COPY scripts/download_model.py .
RUN chmod +x download_model.py && python download_model.py ${WHISPER_MODEL}
COPY scripts/download_models.py .
RUN python download_models.py ${WHISPER_MODELS}
ENTRYPOINT ["watchmedo", "auto-restart", "-d" , "app/worker", "-p", "*.py", "--recursive", "celery", "--", "--app=app.worker.main.celery", "worker", "--loglevel=info", "--concurrency=1"]
CMD watchmedo auto-restart -d app/worker -p *.py --recursive celery -- --app=app.worker.main.celery worker --loglevel=info --concurrency=1