chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #205 from prashanthellina/v3-fix-diarization

Browse Source
This commit is contained in:
Max Bain
2023-04-30 21:08:45 +01:00
committed by GitHub
parent 31a9ec7466 601c91140f
commit e24ca9e0a2
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
whisperx/alignment.py
View File

@ -450,8 +450,8 @@ def align(
"end": srow["end"],
"text": text,
"words": word_list,
# "word-segments": wseg,
# "char-segments": cseg
"word-segments": wseg,
"char-segments": cseg
}
)

6
whisperx/transcribe.py
View File

@ -203,6 +203,12 @@ def cli():
# >> Write
for result, audio_path in results:
# Remove pandas dataframes from result so that
# we can serialize the result with json
for seg in result["segments"]:
seg.pop("word-segments", None)
seg.pop("char-segments", None)
writer(result, audio_path, writer_args)
if __name__ == "__main__":