Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-03-30 14:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix output_dir argument when audio file is a path (#45)

Browse Source
This commit is contained in:
EliEron 2022-09-23 04:38:37 +02:00 committed by GitHub
parent c0607e8d22
commit 759e8d47a8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
whisper/transcribe.py
View File

@ -286,12 +286,14 @@ def cli():
**args,
)
audio_basename = os.path.basename(audio_path)
# save TXT
with open(os.path.join(output_dir, audio_path + ".txt"), "w") as txt:
with open(os.path.join(output_dir, audio_basename + ".txt"), "w") as txt:
print(result["text"], file=txt)
# save VTT
with open(os.path.join(output_dir, audio_path + ".vtt"), "w") as vtt:
with open(os.path.join(output_dir, audio_basename + ".vtt"), "w") as vtt:
write_vtt(result["segments"], file=vtt)