bellingcat/whisperbox-transcribe
1
0
Fork 0
mirror of https://github.com/bellingcat/whisperbox-transcribe.git synced 2026-06-12 21:48:35 +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

6
docker/prod/worker.Dockerfile
View File

@@ -10,7 +10,7 @@ RUN python -m venv /opt/venv && \
FROM python:3.10 as python-deploy
ARG WHISPER_MODEL
ARG WHISPER_MODELS
WORKDIR /etc/whisperbox
@@ -24,7 +24,7 @@ COPY app ./app
ENV VIRTUAL_ENV /opt/venv
ENV PATH /opt/venv/bin:$PATH
COPY scripts/download_model.py .
RUN chmod +x download_model.py && python download_model.py ${WHISPER_MODEL:-small}
COPY scripts/download_models.py .
RUN python download_models.py ${WHISPER_MODELS}
CMD celery --app=app.worker.main.celery worker --loglevel=info --concurrency=1