fixed but where speaker dict included segment informations

This commit is contained in:
Jaikinator
2023-06-19 12:11:28 +02:00
parent d3606a2dab
commit 280cfa3c35
+1 -2
View File
@@ -55,7 +55,7 @@ class Diariser:
### ###
if len(dia_list) == 1: if len(dia_list) == 1:
normalized_output.append([0, 0, dia_list[0]]) normalized_output.append([0, 0, dia_list[0][2]])
else: else:
for i, (_, _, speaker) in enumerate(dia_list): for i, (_, _, speaker) in enumerate(dia_list):
@@ -158,4 +158,3 @@ class Diariser:
def __str__(self): def __str__(self):
return f"Diarisation(model={self.model})" return f"Diarisation(model={self.model})"