From b5dab23dd4cbd3a5b075c50f14de5f22ec622705 Mon Sep 17 00:00:00 2001 From: Jaikinator Date: Mon, 12 Jun 2023 15:54:28 +0200 Subject: [PATCH] diarization in seconds --- autotranscript/diarisation.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/autotranscript/diarisation.py b/autotranscript/diarisation.py index b0c9e84..be5e534 100644 --- a/autotranscript/diarisation.py +++ b/autotranscript/diarisation.py @@ -23,13 +23,9 @@ class Diarisation: :param kwargs: kwargs for diarization model :return: diarization """ - - print(f'Start diarization of audio file: {audiofile}') - + diarization = self.model(audiofile,*args, **kwargs) - print('Diarization finished') - out = self.format_diarization_output(diarization) return out @@ -81,9 +77,8 @@ class Diarisation: current_speaker]) for outp in normalized_output: - #convert in milliseconds - start = dia_list[outp[0]][0].start * 1000 - end = dia_list[outp[1]][0].end * 1000 + start = dia_list[outp[0]][0].start + end = dia_list[outp[1]][0].end diarization_output["segments"].append([start, end]) diarization_output["speakers"].append(outp[2])