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

make aud optional

Browse Source
This commit is contained in:
Max Bain
2023-06-07 11:47:49 +01:00
committed by GitHub
parent 076ff96eb2
commit b13778fefd
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
whisperx/utils.py
View File

@ -399,13 +399,15 @@ def get_writer(
output_format: str, output_dir: str output_format: str, output_dir: str
) -> Callable[[dict, TextIO, dict], None]: ) -> Callable[[dict, TextIO, dict], None]:
writers = { writers = {
"aud": WriteAudacity,
"txt": WriteTXT, "txt": WriteTXT,
"vtt": WriteVTT, "vtt": WriteVTT,
"srt": WriteSRT, "srt": WriteSRT,
"tsv": WriteTSV, "tsv": WriteTSV,
"json": WriteJSON, "json": WriteJSON,
} }
optional_writers = {
"aud": WriteAudacity,
}
if output_format == "all": if output_format == "all":
all_writers = [writer(output_dir) for writer in writers.values()] all_writers = [writer(output_dir) for writer in writers.values()]
@ -416,6 +418,8 @@ def get_writer(
return write_all return write_all
if output_format in optional_writers:
return optional_writers[output_format](output_dir)
return writers[output_format](output_dir) return writers[output_format](output_dir)
def interpolate_nans(x, method='nearest'): def interpolate_nans(x, method='nearest'):