jianbing/whisper repository overview

FROM python:3.9.9


WORKDIR /whisper

RUN echo 'deb https://mirrors.aliyun.com/debian/ bullseye main non-free contrib' > /etc/apt/sources.list \
    && echo 'deb-src https://mirrors.aliyun.com/debian/ bullseye main non-free contrib' >> /etc/apt/sources.list \
    && echo 'deb https://mirrors.aliyun.com/debian-security/ bullseye-security main' >> /etc/apt/sources.list \
    && echo 'deb-src https://mirrors.aliyun.com/debian-security/ bullseye-security main' >> /etc/apt/sources.list \
    && echo 'deb https://mirrors.aliyun.com/debian/ bullseye-updates main non-free contrib' >> /etc/apt/sources.list \
    && echo 'deb-src https://mirrors.aliyun.com/debian/ bullseye-updates main non-free contrib' >> /etc/apt/sources.list \
    && echo 'deb https://mirrors.aliyun.com/debian/ bullseye-backports main non-free contrib' >> /etc/apt/sources.list \
    && echo 'deb-src https://mirrors.aliyun.com/debian/ bullseye-backports main non-free contrib' >> /etc/apt/sources.list \
    && apt update
RUN apt install ffmpeg -y 
RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
RUN pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ git+https://github.com/openai/whisper.git 
# RUN pip install --upgrade --no-deps --force-reinstall git+https://github.com/openai/whisper.git

# COPY ./whisper /whisper/
# RUN python setup.py

COPY ./audio.mp3 /whisper/


# 触发模型下载
RUN whisper audio.mp3 --model tiny
RUN whisper audio.mp3 --model base
# 默认使用--model small 模型进行转换
RUN whisper audio.mp3
RUN whisper audio.mp3 --model medium
RUN whisper audio.mp3 --model large

#  i7-7700 CPU  10s音频 large 用时 3分17秒
#  i7-7700 CPU  10s音频 tiny 用时 10.83秒