Spaces:
Build error
Build error
Commit
·
e491b4f
1
Parent(s):
0edfdd4
add alimeeting model
Browse files
examples.py
CHANGED
|
@@ -58,6 +58,27 @@ examples = [
|
|
| 58 |
4,
|
| 59 |
"./test_wavs/tibetan/a_0_cacm-A70_31117.wav",
|
| 60 |
],
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 61 |
# librispeech
|
| 62 |
# https://huggingface.co/csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless5-2022-05-13/tree/main/test_wavs
|
| 63 |
[
|
|
|
|
| 58 |
4,
|
| 59 |
"./test_wavs/tibetan/a_0_cacm-A70_31117.wav",
|
| 60 |
],
|
| 61 |
+
[
|
| 62 |
+
"Chinese",
|
| 63 |
+
"desh2608/icefall-asr-alimeeting-pruned-transducer-stateless7",
|
| 64 |
+
"greedy_search",
|
| 65 |
+
4,
|
| 66 |
+
"./test_wavs/alimeeting/R8003_M8001-8004-165.wav",
|
| 67 |
+
],
|
| 68 |
+
[
|
| 69 |
+
"Chinese",
|
| 70 |
+
"desh2608/icefall-asr-alimeeting-pruned-transducer-stateless7",
|
| 71 |
+
"greedy_search",
|
| 72 |
+
4,
|
| 73 |
+
"./test_wavs/alimeeting/R8008_M8013-8049-74.wav",
|
| 74 |
+
],
|
| 75 |
+
[
|
| 76 |
+
"Chinese",
|
| 77 |
+
"desh2608/icefall-asr-alimeeting-pruned-transducer-stateless7",
|
| 78 |
+
"greedy_search",
|
| 79 |
+
4,
|
| 80 |
+
"./test_wavs/alimeeting/R8009_M8020_N_SPK8026-8026-209.wav",
|
| 81 |
+
],
|
| 82 |
# librispeech
|
| 83 |
# https://huggingface.co/csukuangfj/icefall-asr-librispeech-pruned-transducer-stateless5-2022-05-13/tree/main/test_wavs
|
| 84 |
[
|
model.py
CHANGED
|
@@ -311,12 +311,18 @@ def _get_alimeeting_pre_trained_model(
|
|
| 311 |
num_active_paths: int,
|
| 312 |
):
|
| 313 |
assert repo_id in [
|
|
|
|
| 314 |
"luomingshuang/icefall_asr_alimeeting_pruned_transducer_stateless2",
|
| 315 |
], repo_id
|
| 316 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 317 |
nn_model = _get_nn_model_filename(
|
| 318 |
repo_id=repo_id,
|
| 319 |
-
filename=
|
| 320 |
)
|
| 321 |
tokens = _get_token_filename(repo_id=repo_id)
|
| 322 |
|
|
@@ -532,6 +538,7 @@ def _get_german_pre_trained_model(
|
|
| 532 |
|
| 533 |
chinese_models = {
|
| 534 |
"luomingshuang/icefall_asr_wenetspeech_pruned_transducer_stateless2": _get_wenetspeech_pre_trained_model, # noqa
|
|
|
|
| 535 |
"yuekai/icefall-asr-aishell2-pruned-transducer-stateless5-A-2022-07-12": _get_aishell2_pretrained_model, # noqa
|
| 536 |
"yuekai/icefall-asr-aishell2-pruned-transducer-stateless5-B-2022-07-12": _get_aishell2_pretrained_model, # noqa
|
| 537 |
"luomingshuang/icefall_asr_aidatatang-200zh_pruned_transducer_stateless2": _get_aidatatang_200zh_pretrained_mode, # noqa
|
|
|
|
| 311 |
num_active_paths: int,
|
| 312 |
):
|
| 313 |
assert repo_id in [
|
| 314 |
+
"desh2608/icefall-asr-alimeeting-pruned-transducer-stateless7",
|
| 315 |
"luomingshuang/icefall_asr_alimeeting_pruned_transducer_stateless2",
|
| 316 |
], repo_id
|
| 317 |
|
| 318 |
+
if repo_id == "desh2608/icefall-asr-alimeeting-pruned-transducer-stateless7":
|
| 319 |
+
filename = "cpu_jit.pt"
|
| 320 |
+
elif repo_id == "luomingshuang/icefall_asr_alimeeting_pruned_transducer_stateless2":
|
| 321 |
+
filename = "cpu_jit_torch_1.7.1.pt"
|
| 322 |
+
|
| 323 |
nn_model = _get_nn_model_filename(
|
| 324 |
repo_id=repo_id,
|
| 325 |
+
filename=filename,
|
| 326 |
)
|
| 327 |
tokens = _get_token_filename(repo_id=repo_id)
|
| 328 |
|
|
|
|
| 538 |
|
| 539 |
chinese_models = {
|
| 540 |
"luomingshuang/icefall_asr_wenetspeech_pruned_transducer_stateless2": _get_wenetspeech_pre_trained_model, # noqa
|
| 541 |
+
"desh2608/icefall-asr-alimeeting-pruned-transducer-stateless7": _get_alimeeting_pre_trained_model,
|
| 542 |
"yuekai/icefall-asr-aishell2-pruned-transducer-stateless5-A-2022-07-12": _get_aishell2_pretrained_model, # noqa
|
| 543 |
"yuekai/icefall-asr-aishell2-pruned-transducer-stateless5-B-2022-07-12": _get_aishell2_pretrained_model, # noqa
|
| 544 |
"luomingshuang/icefall_asr_aidatatang-200zh_pruned_transducer_stateless2": _get_aidatatang_200zh_pretrained_mode, # noqa
|
test_wavs/alimeeting/R8003_M8001-8004-165.wav
ADDED
|
Binary file (525 kB). View file
|
|
|
test_wavs/alimeeting/R8008_M8013-8049-74.wav
ADDED
|
Binary file (241 kB). View file
|
|
|
test_wavs/alimeeting/R8009_M8020_N_SPK8026-8026-209.wav
ADDED
|
Binary file (309 kB). View file
|
|
|
test_wavs/alimeeting/trans.txt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
R8009_M8020_N_SPK8026-8026-209 并不是说一天的话就一定要对一个人进行一个了解这样的话
|
| 2 |
+
R8003_M8001-8004-165 如果他要是不愿意提供地址也不愿意接收礼物那么第二个这个分支可能就省省下了
|
| 3 |
+
R8008_M8013-8049-74 面试的话五月五号到五月十号吧面试
|