fixed bug when only one speaker exists

2023-06-19 11:52:21 +02:00
parent 3ef7353db5
commit 7bfd294bbd
1 changed files with 20 additions and 19 deletions
@@ -54,9 +54,11 @@ class Diariser:
        # This loop removes these duplicates
        ###
        
+        if len(dia_list) == 1:
+            normalized_output.append([0, 0, dia_list[0]])
+        else:
            
            for i, (_, _, speaker) in enumerate(dia_list):
-            
                if i == 0:
                    current_speaker = speaker

@@ -84,7 +86,6 @@ class Diariser:

            diarization_output["segments"].append([start, end])
            diarization_output["speakers"].append(outp[2])
-
        return diarization_output
    
    @staticmethod