kotoba-whisper-v2.2 / pipeline /test_pipeline.py
asahi417's picture
init
9d62abd
raw
history blame contribute delete
523 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)
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))