Whisperでローカル文字起こし

dictation.py

import argparse
import os
from pathlib import Path

import whisper


def main(infile: str) -> None:
    in_path = Path(infile)
    out_path = in_path.with_name(in_path.stem + ".txt")
    model = whisper.load_model("medium")  # base < small < medium < large
    result = model.transcribe(infile, verbose=True, language="ja")
    lines = []
    for segment in result["segments"]:
        lines.append(segment["text"])
    Path(out_path).write_text(os.linesep.join(lines), encoding="utf-8")


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("infile", type=str)
    args = parser.parse_args()
    main(args.infile)