From 280cfa3c35e391c752d2b7b811f214c883c47f81 Mon Sep 17 00:00:00 2001 From: Jaikinator Date: Mon, 19 Jun 2023 12:11:28 +0200 Subject: [PATCH] fixed but where speaker dict included segment informations --- autotranscript/diarisation.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/autotranscript/diarisation.py b/autotranscript/diarisation.py index 931d395..5b71f88 100644 --- a/autotranscript/diarisation.py +++ b/autotranscript/diarisation.py @@ -55,7 +55,7 @@ class Diariser: ### if len(dia_list) == 1: - normalized_output.append([0, 0, dia_list[0]]) + normalized_output.append([0, 0, dia_list[0][2]]) else: for i, (_, _, speaker) in enumerate(dia_list): @@ -158,4 +158,3 @@ class Diariser: def __str__(self): return f"Diarisation(model={self.model})" -