diff --git a/Dockerfile b/Dockerfile index 7a9bb02..0b4d1d2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -9,7 +9,6 @@ RUN apt-get update && apt-get install -y \ python3 \ python3-pip \ ffmpeg \ - wget \ libavcodec-dev \ libavformat-dev \ libavutil-dev \ @@ -21,10 +20,6 @@ RUN git clone https://github.com/ggerganov/whisper.cpp.git /whisper.cpp WORKDIR /whisper.cpp RUN make -# Download model -RUN mkdir -p models && \ - wget https://huggingface.co/datasets/ggerganov/whisper.cpp/resolve/main/ggml-base.en.bin -O models/ggml-base.en.bin - FROM ubuntu:22.04 AS runtime # Install runtime dependencies @@ -38,9 +33,13 @@ RUN apt-get update && apt-get install -y \ libswresample-dev \ && rm -rf /var/lib/apt/lists/* -# Copy only necessary files from builder +# Copy whisper.cpp executable COPY --from=builder /whisper.cpp/main /whisper.cpp/main -COPY --from=builder /whisper.cpp/models /whisper.cpp/models + +# Create models directory and copy your local model +RUN mkdir -p /whisper.cpp/models +COPY model/ggml-tiny.en.bin /whisper.cpp/models/ggml-tiny.en.bin + WORKDIR /whisper.cpp # Install Python dependencies diff --git a/model/ggml-tiny.en.bin b/model/ggml-tiny.en.bin new file mode 100644 index 0000000..9dd1a6b Binary files /dev/null and b/model/ggml-tiny.en.bin differ