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

Update

Browse Source
This commit is contained in:
litagin02
2024-02-25 13:15:40 +09:00
parent e76dca6fbb
commit 8f27c79219
9 changed files with 46 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
transcribe.py
View File

@@ -78,12 +78,9 @@ if __name__ == "__main__":
language_id = Languages.ZH.value
else:
raise ValueError(f"{language} is not supported.")
with open(output_file, "w", encoding="utf-8") as f:
for wav_file in tqdm(wav_files, file=SAFE_STDOUT):
file_name = os.path.basename(wav_file)
text = transcribe(
wav_file, initial_prompt=initial_prompt, language=language
)
for wav_file in tqdm(wav_files, file=SAFE_STDOUT):
file_name = os.path.basename(wav_file)
text = transcribe(wav_file, initial_prompt=initial_prompt, language=language)
with open(output_file, "a", encoding="utf-8") as f:
f.write(f"{file_name}|{model_name}|{language_id}|{text}\n")
f.flush()
sys.exit(0)