| # docker build -t whisper-webui --build-arg WHISPER_IMPLEMENTATION=whisper . | |
| FROM huggingface/transformers-pytorch-gpu | |
| EXPOSE 7860 | |
| ARG WHISPER_IMPLEMENTATION=faster-whisper | |
| ENV WHISPER_IMPLEMENTATION=${WHISPER_IMPLEMENTATION} | |
| ADD . /opt/whisper-webui/ | |
| # Latest version of transformers-pytorch-gpu seems to lack tk. | |
| # Further, pip install fails, so we must upgrade pip first. | |
| RUN apt-get -y install python3-tk | |
| RUN python3 -m pip install --upgrade pip | |
| RUN if [ "${WHISPER_IMPLEMENTATION}" = "whisper" ]; then \ | |
| python3 -m pip install -r /opt/whisper-webui/requirements-whisper.txt; \ | |
| else \ | |
| python3 -m pip install -r /opt/whisper-webui/requirements-fasterWhisper.txt; \ | |
| fi | |
| # Note: Models will be downloaded on demand to the directory /root/.cache/whisper. | |
| # You can also bind this directory in the container to somewhere on the host. | |
| # To be able to see logs in real time | |
| ENV PYTHONUNBUFFERED=1 | |
| WORKDIR /opt/whisper-webui/ | |
| ENTRYPOINT ["python3"] | |
| CMD ["app.py", "--whisper_implementation", "faster-whisper", "--input_audio_max_duration", "-1", "--server_name", "0.0.0.0", "--auto_parallel", "True"] |