"""wget https://huggingface.co/kotoba-tech/kotoba-whisper-v2.2/resolve/main/sample_audio/sample_diarization_japanese.mp3"""
from pprint import pprint
from transformers.pipelines import pipeline

pipe = pipeline(model="kotoba-tech/kotoba-whisper-v2.2", chunk_length_s=None, batch_size=16, trust_remote_code=True)
pprint(pipe("sample_diarization_japanese.mp3"))
pprint(pipe("sample_diarization_japanese.mp3", add_punctuation=True))
pprint(pipe("sample_diarization_japanese.mp3", add_silence_end=0.5, add_silence_start=0.5))