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

tomchang25 · Sep 23, 2022 · 759e8d4 · 759e8d4
1 parent c0607e8
commit 759e8d4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/whisper/transcribe.py b/whisper/transcribe.py
@@ -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)