From e721c09b87eeaacaba37ba6bb804ec13a2aa7e49 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Sat, 27 Apr 2024 12:59:47 +0900 Subject: [PATCH] Fix filename for transcribe to use relative path instead of name --- transcribe.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/transcribe.py b/transcribe.py index 210ce2c..898fdc9 100644 --- a/transcribe.py +++ b/transcribe.py @@ -183,8 +183,9 @@ if __name__ == "__main__": num_beams=num_beams, no_repeat_ngram_size=no_repeat_ngram_size, ) + wav_rel_path = wav_file.relative_to(input_dir) with open(output_file, "a", encoding="utf-8") as f: - f.write(f"{wav_file.name}|{model_name}|{language_id}|{text}\n") + f.write(f"{wav_rel_path}|{model_name}|{language_id}|{text}\n") else: model_id = f"openai/whisper-{args.model}" logger.info(f"Loading HF Whisper model ({model_id})") @@ -202,6 +203,7 @@ if __name__ == "__main__": ) with open(output_file, "w", encoding="utf-8") as f: for wav_file, text in zip(wav_files, results): - f.write(f"{wav_file.name}|{model_name}|{language_id}|{text}\n") + wav_rel_path = wav_file.relative_to(input_dir) + f.write(f"{wav_rel_path}|{model_name}|{language_id}|{text}\n") sys.exit(0)