admin/scribe
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

diarization in seconds

Browse Source
This commit is contained in:
Jaikinator
2023-06-12 15:54:28 +02:00
parent ca42d631cd
commit b5dab23dd4
1 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
autotranscript/diarisation.py
+2 -7
View File
@@ -24,12 +24,8 @@ class Diarisation:
:return: diarization :return: diarization
""" """
print(f'Start diarization of audio file: {audiofile}')
diarization = self.model(audiofile,*args, **kwargs) diarization = self.model(audiofile,*args, **kwargs)
print('Diarization finished')
out = self.format_diarization_output(diarization) out = self.format_diarization_output(diarization)
return out return out
@@ -81,9 +77,8 @@ class Diarisation:
current_speaker]) current_speaker])
for outp in normalized_output: for outp in normalized_output:
#convert in milliseconds start = dia_list[outp[0]][0].start
start = dia_list[outp[0]][0].start * 1000 end = dia_list[outp[1]][0].end
end = dia_list[outp[1]][0].end * 1000
diarization_output["segments"].append([start, end]) diarization_output["segments"].append([start, end])
diarization_output["speakers"].append(outp[2]) diarization_output["speakers"].append(outp[2])