56 lines
1.1 KiB
Docker
56 lines
1.1 KiB
Docker
FROM ubuntu:22.04 AS builder
|
|
|
|
# Install build dependencies
|
|
RUN apt-get update && apt-get install -y \
|
|
git \
|
|
cmake \
|
|
make \
|
|
g++ \
|
|
python3 \
|
|
python3-pip \
|
|
ffmpeg \
|
|
libavcodec-dev \
|
|
libavformat-dev \
|
|
libavutil-dev \
|
|
libswresample-dev \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Clone and build whisper.cpp
|
|
RUN git clone https://github.com/ggerganov/whisper.cpp.git /whisper.cpp
|
|
WORKDIR /whisper.cpp
|
|
RUN make
|
|
|
|
FROM ubuntu:22.04 AS runtime
|
|
|
|
# Install runtime dependencies
|
|
RUN apt-get update && apt-get install -y \
|
|
python3 \
|
|
python3-pip \
|
|
ffmpeg \
|
|
libavcodec-dev \
|
|
libavformat-dev \
|
|
libavutil-dev \
|
|
libswresample-dev \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Copy whisper.cpp executable
|
|
COPY --from=builder /whisper.cpp/main /whisper.cpp/main
|
|
|
|
# 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
|
|
RUN pip3 install flask flask-cors
|
|
|
|
# Copy API server script
|
|
COPY app.py .
|
|
|
|
# Expose port
|
|
EXPOSE 4002
|
|
|
|
# Run the server
|
|
CMD ["python3", "app.py"]
|