fixed but where speaker dict included segment informations

2023-06-19 12:11:28 +02:00
parent d3606a2dab
commit 280cfa3c35
1 changed files with 1 additions and 2 deletions
@@ -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})"
-