diff --git a/verbanote/process.py b/verbanote/process.py index eb36eee..809aa16 100644 --- a/verbanote/process.py +++ b/verbanote/process.py @@ -20,15 +20,14 @@ class TxtTranscription: def diarize(audiofile: Path, pipeline: Pipeline, output_path: Path) -> Path: audiofile_prepended = _add_audio_silence(audiofile) + logging.info(f"Beginning diarization of {audiofile}...") DIARIZE_FILE = {"uri": "not-important", "audio": audiofile_prepended} dz = pipeline(DIARIZE_FILE) out_file = Path.joinpath(output_path, "diarization.txt") with open(out_file, "w") as text_file: text_file.write(str(dz)) - - print("Diarized:") - print(*list(dz.itertracks(yield_label=True))[:10], sep="\n") + logging.info(f"Created diarization in {out_file}.") return out_file