init

pipeline/test_pipeline.py

@@ -3,8 +3,7 @@ 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)
+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))
 
-output = pipe("sample_diarization_japanese.mp3", add_punctu)
-pprint(output)