From 673fa673b6ba60737598d2ceb0d86283a745f371 Mon Sep 17 00:00:00 2001
From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com>
Date: Tue, 16 Jan 2024 17:35:12 +0800
Subject: [PATCH] Add files via upload

---
 tools/damo_asr/cmd-asr.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 tools/damo_asr/cmd-asr.py

diff --git a/tools/damo_asr/cmd-asr.py b/tools/damo_asr/cmd-asr.py
new file mode 100644
index 0000000..9d1c4b9
--- /dev/null
+++ b/tools/damo_asr/cmd-asr.py
@@ -0,0 +1,27 @@
+# -*- coding:utf-8 -*-
+
+from modelscope.pipelines import pipeline
+from modelscope.utils.constant import Tasks
+import sys,os,traceback
+dir=sys.argv[1]
+# opt_name=dir.split("\\")[-1].split("/")[-1]
+opt_name=os.path.basename(dir)
+inference_pipeline = pipeline(
+    task=Tasks.auto_speech_recognition,
+    model='tools/damo_asr/models/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch',
+    vad_model='tools/damo_asr/models/speech_fsmn_vad_zh-cn-16k-common-pytorch',
+    punc_model='tools/damo_asr/models/punc_ct-transformer_zh-cn-common-vocab272727-pytorch',
+)
+
+opt=[]
+for name in os.listdir(dir):
+    try:
+        text = inference_pipeline(audio_in="%s/%s"%(dir,name))["text"]
+        opt.append("%s/%s|%s|ZH|%s"%(dir,name,opt_name,text))
+    except:
+        print(traceback.format_exc())
+
+opt_dir="output/asr_opt"
+os.makedirs(opt_dir,exist_ok=True)
+with open("%s/%s.list"%(opt_dir,opt_name),"w",encoding="utf-8")as f:f.write("\n".join(opt))
+