from pprint import pprint
from transformers.pipelines import pipeline

test_audio = "/Users/asahiu/Desktop/speaker_diariazation_sample_1.wav"
pipe = pipeline(model="kotoba-tech/kotoba-whisper-v2.2", chunk_length_s=15, batch_size=16, trust_remote_code=True)
output = pipe(test_audio)
pprint(output)