File size: 1,056 Bytes
2698c96 a14ac5a 2698c96 a14ac5a 596f6a6 2698c96 a14ac5a 0d8d833 a14ac5a 60dc82d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
# docker build -t whisper-webui --build-arg WHISPER_IMPLEMENTATION=whisper .
FROM huggingface/transformers-pytorch-gpu
EXPOSE 7860
ARG WHISPER_IMPLEMENTATION=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.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", "--input_audio_max_duration", "-1", "--server_name", "0.0.0.0", "--auto_parallel", "True"] |