Files
whisperbox-transcribe/docker/prod/worker.Dockerfile
2023-02-28 21:40:55 +01:00

31 lines
731 B
Docker

FROM python:3.10 AS python-build
WORKDIR /etc/whisperbox
COPY pyproject.toml .
RUN python -m venv /opt/venv && \
/opt/venv/bin/pip install -U pip wheel && \
/opt/venv/bin/pip install -U .[worker]
FROM python:3.10 as python-deploy
ARG WHISPER_MODEL
WORKDIR /etc/whisperbox
COPY --from=python-build /opt/venv /opt/venv
COPY --from=mwader/static-ffmpeg:latest /ffmpeg /usr/local/bin/
COPY --from=mwader/static-ffmpeg:latest /ffprobe /usr/local/bin/
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}
CMD celery --app=app.worker.main.celery worker --loglevel=info --concurrency=1