Adjust-asr

This commit is contained in:
KamioRinn
2024-02-11 01:34:25 +08:00
parent 41041715a4
commit ccc7757880
3 changed files with 20 additions and 5 deletions

View File

@@ -23,6 +23,14 @@ model = AutoModel(
punc_model_revision = "v2.0.4",
)
def only_asr(input_file):
try:
text = model.generate(input=input_file)[0]["text"]
except:
text = ''
print(traceback.format_exc())
return text
def execute_asr(input_folder, output_folder, model_size, language):
input_file_names = os.listdir(input_folder)
input_file_names.sort()