aadnk commited on
Commit
596f6a6
1 Parent(s): 49e2986

Fix latest transformer docker image

Browse files

The docker image after 4.23.1 seems to contain a broken installation
of PIP, and it's also lacking python3-tk. This commit ensures these
are updated and installed before the rest of the dependencies.

Files changed (1) hide show
  1. dockerfile +6 -1
dockerfile CHANGED
@@ -2,7 +2,12 @@ FROM huggingface/transformers-pytorch-gpu
2
  EXPOSE 7860
3
 
4
  ADD . /opt/whisper-webui/
5
- RUN python3 -m pip install -r /opt/whisper-webui/requirements.txt
 
 
 
 
 
6
 
7
  # Note: Models will be downloaded on demand to the directory /root/.cache/whisper.
8
  # You can also bind this directory in the container to somewhere on the host.
 
2
  EXPOSE 7860
3
 
4
  ADD . /opt/whisper-webui/
5
+
6
+ # Latest version of transformers-pytorch-gpu seems to lack tk.
7
+ # Further, pip install fails, so we must upgrade pip first.
8
+ RUN apt-get -y install python3-tk
9
+ RUN python3 -m pip install --upgrade pip &&\
10
+ python3 -m pip install -r /opt/whisper-webui/requirements.txt
11
 
12
  # Note: Models will be downloaded on demand to the directory /root/.cache/whisper.
13
  # You can also bind this directory in the container to somewhere on the host.