This commit is contained in:
XXXXRT666 2024-07-01 23:14:13 +08:00 committed by GitHub
parent 1ff74f001a
commit 65d754965e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 3 deletions

View File

@ -78,7 +78,7 @@ def execute_asr(input_folder, output_folder, model_size, language, precision):
text += segment.text text += segment.text
output.append(f"{file_path}|{output_file_name}|{info.language.upper()}|{text}") output.append(f"{file_path}|{output_file_name}|{info.language.upper()}|{text}")
except: except:
return print(traceback.format_exc()) print(traceback.format_exc())
output_folder = output_folder or "output/asr_opt" output_folder = output_folder or "output/asr_opt"
os.makedirs(output_folder, exist_ok=True) os.makedirs(output_folder, exist_ok=True)

View File

@ -1,5 +1,5 @@
import os,argparse import os,argparse
import traceback
from modelscope.pipelines import pipeline from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks from modelscope.utils.constant import Tasks
from tqdm import tqdm from tqdm import tqdm
@ -12,7 +12,10 @@ def execute_denoise(input_folder,output_folder):
# print(input_folder) # print(input_folder)
# print(list(os.listdir(input_folder).sort())) # print(list(os.listdir(input_folder).sort()))
for name in tqdm(os.listdir(input_folder)): for name in tqdm(os.listdir(input_folder)):
ans("%s/%s"%(input_folder,name),output_path='%s/%s'%(output_folder,name)) try:
ans("%s/%s"%(input_folder,name),output_path='%s/%s'%(output_folder,name))
except:
traceback.print_exc()
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()