aadnk's picture
Fix latest transformer docker image
596f6a6
raw
history blame
651 Bytes
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"]