# 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", "--share", "True"]