Spaces:
Running
Running
add bilingual small model (new default)
Browse files- app/asr_worker.py +9 -0
- 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 },
|