teoha commited on
Commit
d5e269e
·
1 Parent(s): 0984899

Harcoded model type as base

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -6
Dockerfile CHANGED
@@ -6,8 +6,7 @@ RUN mkdir /whisper && \
6
 
7
  WORKDIR /whisper
8
 
9
- ARG model=base
10
- RUN bash ./models/download-ggml-model.sh "${model}"
11
  RUN make main
12
 
13
  FROM python:3.10 as whisper
@@ -23,11 +22,9 @@ COPY main.py /root
23
  COPY youtubeaudio.py /root
24
 
25
  WORKDIR /root
26
- ARG model=base
27
- ENV model=$model
28
  RUN mkdir /root/models
29
- RUN mkdir -p -m 777 /tmp/holosubs/results
30
- COPY --from=build "/whisper/models/ggml-${model}.bin" "/root/models/ggml-${model}.bin"
31
  COPY --from=build /whisper/main /usr/local/bin/whisper
32
 
33
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 
6
 
7
  WORKDIR /whisper
8
 
9
+ RUN bash ./models/download-ggml-model.sh "base"
 
10
  RUN make main
11
 
12
  FROM python:3.10 as whisper
 
22
  COPY youtubeaudio.py /root
23
 
24
  WORKDIR /root
25
+ ENV model=base
 
26
  RUN mkdir /root/models
27
+ COPY --from=build "/whisper/models/ggml-base.bin" "/root/models/ggml-base.bin"
 
28
  COPY --from=build /whisper/main /usr/local/bin/whisper
29
 
30
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]