rusty-whisper

Rust 实现 Whisper。有关模型的更多信息：https://github.com/openai/whisper

该软件包基于 tract。

快速开始

下载 权重 和 示例音频文件 并运行简单的推理代码

use rusty_whisper::Whisper;

fn main() {
    let whisper = Whisper::new(
        "weights/encoder.onnx",
        "weights/decoder.onnx",
        "weights/multilingual.tiktoken",
        "weights/positional_embedding.npz",
        "weights/mel_filters.npz",
    );
    let result = whisper.recognize_from_audio("data/audio.wav");
    println!("{}", result);
}

该模型仅与 16 位 WAV 文件兼容，因此请在运行工具之前确保将输入转换。例如，您可以使用 ffmpeg 如下所示

ffmpeg -i input.mp3 -ar 16000 -ac 1 -c:a pcm_s16le output.wav