File size: 1,297 Bytes
a97f2c4
 
0755ca5
a97f2c4
 
e40ee50
a97f2c4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4174395
 
 
a97f2c4
 
 
 
 
 
 
 
 
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
31
32
33
# 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

RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8_8.5.0.96-1+cuda11.7_amd64.deb && \
  dpkg -i libcudnn8_8.5.0.96-1+cuda11.7_amd64.deb && rm libcudnn8_8.5.0.96-1+cuda11.7_amd64.deb

# 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"]