FROM huggingface/transformers-pytorch-gpu EXPOSE 7860 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 &&\ python3 -m pip install -r /opt/whisper-webui/requirements.txt # 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"]