fix model path, add example mp3

.gitattributes

@@ -34,4 +34,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 sensevoice.axmodel filter=lfs diff=lfs merge=lfs -text
-*.axmodel filter=lfs diff=lfs merge=lfs -textdemo.wav filter=lfs diff=lfs merge=lfs -text
+*.axmodel filter=lfs diff=lfs merge=lfs -text
+*.wav filter=lfs diff=lfs merge=lfs -text
+*.mp3 filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+__pycache__

demo.wav → example/en.mp3

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cac60b031ef6ecc85a92b9d6902a1be34bdd9d724de5c21f5f64090695564b38
-size 336428
+oid sha256:f10378336a4e584f3f63799e62f99d5add3c2a401b51d3abe7d3a3a82f255ada
+size 57441

example/ja.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:496dbc43b289e1d0d0cb916df9737450bca56acd8aaca046a7a2472363b1be53
+size 57837

example/ko.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8612f62db8319a6cb4ab4b1d2039bfc32f174f89611889ddafdeb5c0a6070b5f
+size 27909

example/yue.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5098eebc13530a66e4eac1f30d3246e65c9cfc4e096665f9d395aca8eff0d181
+size 31246

example/zh.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0e64de19e4ff9a02e682955c9112f32d2317cfdbb5bc2f3504664044c993f195
+size 44973

gradio_demo.py

@@ -8,9 +8,8 @@ from download_utils import download_model
 use_itn = True # 标点符号预测
 max_len = 68
 
-model_path_root = download_model("SenseVoice")
-model_path = os.path.join(model_path_root, "sensevoice_ax650", "sensevoice.axmodel")
-bpemodel = os.path.join(model_path_root, "chn_jpn_yue_eng_ko_spectok.bpe.model")
+model_path = os.path.join("sensevoice_ax650", "sensevoice.axmodel")
+bpemodel = "chn_jpn_yue_eng_ko_spectok.bpe.model"
 
 assert os.path.exists(model_path), f"model {model_path} not exist"
 

main.py

@@ -21,9 +21,8 @@ def main():
     use_itn = True # 标点符号预测
     max_len = 256
 
-    model_path_root = download_model("SenseVoice")
-    model_path = os.path.join(model_path_root, "sensevoice_ax650", "sensevoice.axmodel")
-    bpemodel = os.path.join(model_path_root, "chn_jpn_yue_eng_ko_spectok.bpe.model")
+    model_path = os.path.join("sensevoice_ax650", "sensevoice.axmodel")
+    bpemodel = "chn_jpn_yue_eng_ko_spectok.bpe.model"
 
     assert os.path.exists(model_path), f"model {model_path} not exist"
 

server.py

@@ -34,9 +34,8 @@ async def load_model():
         use_itn = True # 标点符号预测
         max_len = 68
 
-        model_path_root = download_model("SenseVoice")
-        model_path = os.path.join(model_path_root, "sensevoice_ax650", "sensevoice.axmodel")
-        bpemodel = os.path.join(model_path_root, "chn_jpn_yue_eng_ko_spectok.bpe.model")
+        model_path = os.path.join("sensevoice_ax650", "sensevoice.axmodel")
+        bpemodel = "chn_jpn_yue_eng_ko_spectok.bpe.model"
 
         assert os.path.exists(model_path), f"model {model_path} not exist"
 

test_wer.py

@@ -218,10 +218,9 @@ def main():
     else:
         raise ValueError(f"Unknown dataset type {dataset_type}")
 
-    model_path_root = download_model("SenseVoice")
-    # model_path = os.path.join(model_path_root, "sensevoice_ax650", "sensevoice.axmodel")
-    model_path = "./model_convert/output_dir/model.onnx"
-    bpemodel = os.path.join(model_path_root, "chn_jpn_yue_eng_ko_spectok.bpe.model")
+    # model_path_root = download_model("SenseVoice")
+    model_path = os.path.join("sensevoice_ax650", "sensevoice.axmodel")
+    bpemodel = "chn_jpn_yue_eng_ko_spectok.bpe.model"
 
     assert os.path.exists(model_path), f"model {model_path} not exist"