diff --git a/dockerfile b/dockerfile index 7a29207..378f9fe 100644 --- a/dockerfile +++ b/dockerfile @@ -5,6 +5,8 @@ COPY notescriber /app/notescriber COPY notescriber.py /app/notescriber.py COPY requirements.txt /app/requirements.txt COPY .config.yaml /app/.config.yaml +COPY load_model.py /app/load_model.py WORKDIR /app -RUN pip install -r requirements.txt \ No newline at end of file +RUN pip install -r requirements.txt +RUN /app/load_model.py \ No newline at end of file diff --git a/load_model.py b/load_model.py new file mode 100755 index 0000000..274aac4 --- /dev/null +++ b/load_model.py @@ -0,0 +1,3 @@ +#!/usr/bin/env python +import whisper +model = whisper.load_model("base") \ No newline at end of file