tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fix filename for transcribe to use relative path instead of name

Browse Source
This commit is contained in:
litagin02
2024-04-27 12:59:47 +09:00
parent 9df344b995
commit e721c09b87
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
transcribe.py
View File

@@ -183,8 +183,9 @@ if __name__ == "__main__":
num_beams=num_beams, num_beams=num_beams,
no_repeat_ngram_size=no_repeat_ngram_size, 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: 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: else:
model_id = f"openai/whisper-{args.model}" model_id = f"openai/whisper-{args.model}"
logger.info(f"Loading HF Whisper model ({model_id})") 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: with open(output_file, "w", encoding="utf-8") as f:
for wav_file, text in zip(wav_files, results): 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) sys.exit(0)