fixed bug on which the last speaker diarization where cut of

2023-09-04 12:59:51 +02:00
parent 9bc43b9b18
commit 2643d7999b
1 changed files with 6 additions and 3 deletions
@@ -111,6 +111,7 @@ class Diariser:
        else:
            
            for i, (_, _, speaker) in enumerate(dia_list):
+                
                if i == 0:
                    current_speaker = speaker
                
@@ -125,9 +126,11 @@ class Diariser:
                    index_start_speaker = i
                    current_speaker = speaker

-                if i == len(diarization_output["speakers"]) - 1:
+            
+                if i == len(dia_list) - 1:

                    index_end_speaker = i
+                    
                    normalized_output.append([index_start_speaker, 
                                            index_end_speaker, 
                                            current_speaker])