Luigi commited on
Commit
9fba0a6
·
1 Parent(s): 5f6cff2

add bilingual small model (new default)

Browse files
Files changed (2) hide show
  1. app/asr_worker.py +9 -0
  2. app/static/index.html +2 -0
app/asr_worker.py CHANGED
@@ -14,6 +14,15 @@ converter = OpenCC('s2t')
14
 
15
  # Streaming Zipformer model registry: paths relative to repo root
16
  STREAMING_ZIPFORMER_MODELS = {
 
 
 
 
 
 
 
 
 
17
  "pfluo/k2fsa-zipformer-chinese-english-mixed": {
18
  "tokens": "data/lang_char_bpe/tokens.txt",
19
  "encoder_fp32": "exp/encoder-epoch-99-avg-1.onnx",
 
14
 
15
  # Streaming Zipformer model registry: paths relative to repo root
16
  STREAMING_ZIPFORMER_MODELS = {
17
+ "csukuangfj/k2fsa-zipformer-bilingual-zh-en-t": {
18
+ "tokens": "data/lang_char_bpe/tokens.txt",
19
+ "encoder_fp32": "exp/96/encoder-epoch-99-avg-1.onnx",
20
+ "encoder_int8": "exp/96/encoder-epoch-99-avg-1.int8.onnx",
21
+ "decoder_fp32": "exp/96/decoder-epoch-99-avg-1.onnx",
22
+ "decoder_int8": "exp/96/decoder-epoch-99-avg-1.int8.onnx",
23
+ "joiner_fp32": "exp/96/joiner-epoch-99-avg-1.onnx",
24
+ "joiner_int8": "exp/96/joiner-epoch-99-avg-1.int8.onnx",
25
+ },
26
  "pfluo/k2fsa-zipformer-chinese-english-mixed": {
27
  "tokens": "data/lang_char_bpe/tokens.txt",
28
  "encoder_fp32": "exp/encoder-epoch-99-avg-1.onnx",
app/static/index.html CHANGED
@@ -99,6 +99,7 @@
99
  <div class="controls">
100
  <label for="modelSelect">Model:</label>
101
  <select id="modelSelect">
 
102
  <option value="pfluo/k2fsa-zipformer-chinese-english-mixed">k2fsa-chinese-english-mixed</option>
103
  <option value="k2-fsa/sherpa-onnx-streaming-zipformer-korean-2024-06-16">sherpa-onnx-zipformer-korean</option>
104
  <option value="k2-fsa/sherpa-onnx-streaming-zipformer-multi-zh-hans-2023-12-12">zipformer-multi-zh-hans</option>
@@ -131,6 +132,7 @@
131
 
132
  <script>
133
  const MODEL_METADATA = {
 
134
  "pfluo/k2fsa-zipformer-chinese-english-mixed": { language: ["zh", "en"], size: 342 },
135
  "k2-fsa/sherpa-onnx-streaming-zipformer-korean-2024-06-16": { language: "korean", size: 300 },
136
  "k2-fsa/sherpa-onnx-streaming-zipformer-multi-zh-hans-2023-12-12": { language: "zh-Hans", size: 258 },
 
99
  <div class="controls">
100
  <label for="modelSelect">Model:</label>
101
  <select id="modelSelect">
102
+ <option value="csukuangfj/k2fsa-zipformer-bilingual-zh-en-t">k2fsa-small-bilingual-zh-en</option>
103
  <option value="pfluo/k2fsa-zipformer-chinese-english-mixed">k2fsa-chinese-english-mixed</option>
104
  <option value="k2-fsa/sherpa-onnx-streaming-zipformer-korean-2024-06-16">sherpa-onnx-zipformer-korean</option>
105
  <option value="k2-fsa/sherpa-onnx-streaming-zipformer-multi-zh-hans-2023-12-12">zipformer-multi-zh-hans</option>
 
132
 
133
  <script>
134
  const MODEL_METADATA = {
135
+ "csukuangfj/k2fsa-zipformer-bilingual-zh-en-t": { language: ["zh", "en"], size: 115 },
136
  "pfluo/k2fsa-zipformer-chinese-english-mixed": { language: ["zh", "en"], size: 342 },
137
  "k2-fsa/sherpa-onnx-streaming-zipformer-korean-2024-06-16": { language: "korean", size: 300 },
138
  "k2-fsa/sherpa-onnx-streaming-zipformer-multi-zh-hans-2023-12-12": { language: "zh-Hans", size: 258 },