HoneyTian commited on
Commit
59cce29
1 Parent(s): f819ea9
Files changed (2) hide show
  1. Dockerfile +0 -2
  2. toolbox/k2_sherpa/nn_models.py +35 -1
Dockerfile CHANGED
@@ -19,8 +19,6 @@ ENV HOME=/home/user \
19
  PATH=/home/user/.local/bin:$PATH
20
 
21
  ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.8/site-packages/k2/lib/
22
- #ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/user/.local/lib/python3.8/site-packages/k2/lib/
23
-
24
 
25
  WORKDIR $HOME/app
26
 
 
19
  PATH=/home/user/.local/bin:$PATH
20
 
21
  ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.8/site-packages/k2/lib/
 
 
22
 
23
  WORKDIR $HOME/app
24
 
toolbox/k2_sherpa/nn_models.py CHANGED
@@ -24,13 +24,47 @@ model_map = {
24
  "sub_folder": ".",
25
  "loader": "load_sherpa_offline_recognizer",
26
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
  {
28
  "repo_id": "csukuangfj/sherpa-onnx-paraformer-zh-2023-03-28",
29
  "nn_model_file": "model.int8.onnx",
30
  "tokens_file": "tokens.txt",
31
  "sub_folder": ".",
32
  "loader": "load_sherpa_offline_recognizer_from_paraformer",
33
- }
 
 
 
 
 
 
 
 
 
34
  ]
35
  }
36
 
 
24
  "sub_folder": ".",
25
  "loader": "load_sherpa_offline_recognizer",
26
  },
27
+ {
28
+ "repo_id": "csukuangfj/sherpa-onnx-paraformer-zh-2024-03-09",
29
+ "nn_model_file": "model.int8.onnx",
30
+ "tokens_file": "tokens.txt",
31
+ "sub_folder": ".",
32
+ "loader": "load_sherpa_offline_recognizer_from_paraformer",
33
+ },
34
+ {
35
+ "repo_id": "csukuangfj/sherpa-onnx-paraformer-zh-small-2024-03-09",
36
+ "nn_model_file": "model.int8.onnx",
37
+ "tokens_file": "tokens.txt",
38
+ "sub_folder": ".",
39
+ "loader": "load_sherpa_offline_recognizer_from_paraformer",
40
+ },
41
+ ],
42
+ "English": [
43
+ {
44
+ "repo_id": "csukuangfj/sherpa-onnx-paraformer-en-2024-03-09",
45
+ "nn_model_file": "model.int8.onnx",
46
+ "tokens_file": "tokens.txt",
47
+ "sub_folder": ".",
48
+ "loader": "load_sherpa_offline_recognizer_from_paraformer",
49
+ },
50
+ ],
51
+ "Chinese+English": [
52
  {
53
  "repo_id": "csukuangfj/sherpa-onnx-paraformer-zh-2023-03-28",
54
  "nn_model_file": "model.int8.onnx",
55
  "tokens_file": "tokens.txt",
56
  "sub_folder": ".",
57
  "loader": "load_sherpa_offline_recognizer_from_paraformer",
58
+ },
59
+ ],
60
+ "Chinese+Cantonese+English": [
61
+ {
62
+ "repo_id": "csukuangfj/sherpa-onnx-paraformer-trilingual-zh-cantonese-en",
63
+ "nn_model_file": "model.int8.onnx",
64
+ "tokens_file": "tokens.txt",
65
+ "sub_folder": ".",
66
+ "loader": "load_sherpa_offline_recognizer_from_paraformer",
67
+ },
68
  ]
69
  }
70