kotoba-whisper-v2.2 / pipeline /test_pipeline.py
asahi417's picture
init
f9f9f36
raw
history blame
453 Bytes
"""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)
output = pipe("sample_diarization_japanese.mp3")
pprint(output)
output = pipe("sample_diarization_japanese.mp3", add_punctu)
pprint(output)