Skip to content

Commit

Permalink
medium
Browse files Browse the repository at this point in the history
  • Loading branch information
jwijffels committed May 22, 2024
1 parent d385458 commit 3194c18
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion whisper/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -55,11 +55,12 @@ RUN git clone https://github.com/bnosac/audio.whisper.git -b 0.4.1 \
&& rm -r audio.whisper

## Set the APP_DIR folder + download the medium model by default
ENV WHISPER_MODEL=${WHISPER_MODEL}
ENV APP_DIR=/usr/local/src/whisper
WORKDIR $APP_DIR
RUN mkdir $APP_DIR/models
ENV WHISPER_MODEL_DIR=$APP_DIR/models
ENV WHISPER_MODEL=$WHISPER_MODEL
RUN R -e "cat(sprintf('Downloading <%s> to %s', Sys.getenv('WHISPER_MODEL', unset = 'medium'), Sys.getenv('WHISPER_MODEL_DIR')))"
RUN R -e "audio.whisper::whisper_download_model(Sys.getenv('WHISPER_MODEL', unset = 'medium'), version = '1.5.4')"

CMD ["R"]

0 comments on commit 3194c18

Please sign in to comment.