diarization in seconds

2023-06-12 15:54:28 +02:00
parent ca42d631cd
commit b5dab23dd4
1 changed files with 3 additions and 8 deletions
@@ -24,12 +24,8 @@ class Diarisation:
        :return: diarization
        """
        
-        print(f'Start diarization of audio file: {audiofile}')
-
        diarization = self.model(audiofile,*args, **kwargs)

-        print('Diarization finished')
-
        out = self.format_diarization_output(diarization)

        return out
@@ -81,9 +77,8 @@ class Diarisation:
                                          current_speaker])
       
        for outp in normalized_output:
-            #convert in milliseconds
-            start =  dia_list[outp[0]][0].start * 1000
-            end =  dia_list[outp[1]][0].end * 1000
+            start =  dia_list[outp[0]][0].start 
+            end =  dia_list[outp[1]][0].end

            diarization_output["segments"].append([start, end])
            diarization_output["speakers"].append(outp[2])