| 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-network.py"] |