Commit
·
46f8085
1
Parent(s):
7fd804c
add more models
Browse files
model.py
CHANGED
@@ -214,6 +214,10 @@ def get_pretrained_model(
|
|
214 |
return french_models[repo_id](
|
215 |
repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
|
216 |
)
|
|
|
|
|
|
|
|
|
217 |
elif repo_id in japanese_models:
|
218 |
return japanese_models[repo_id](
|
219 |
repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
|
@@ -1587,9 +1591,15 @@ def _get_streaming_zipformer_pre_trained_model(
|
|
1587 |
"csukuangfj/sherpa-onnx-streaming-zipformer-bilingual-zh-en-2023-02-20",
|
1588 |
"k2-fsa/sherpa-onnx-streaming-zipformer-korean-2024-06-16",
|
1589 |
"csukuangfj/sherpa-onnx-streaming-zipformer-en-kroko-2025-08-06",
|
|
|
|
|
1590 |
], repo_id
|
1591 |
|
1592 |
-
if repo_id
|
|
|
|
|
|
|
|
|
1593 |
encoder_model = _get_nn_model_filename(
|
1594 |
repo_id=repo_id,
|
1595 |
filename="encoder.onnx",
|
@@ -2112,6 +2122,11 @@ german_models = {
|
|
2112 |
|
2113 |
french_models = {
|
2114 |
"shaojieli/sherpa-onnx-streaming-zipformer-fr-2023-04-14": _get_french_pre_trained_model,
|
|
|
|
|
|
|
|
|
|
|
2115 |
}
|
2116 |
|
2117 |
japanese_models = {
|
@@ -2177,6 +2192,7 @@ all_models = {
|
|
2177 |
**arabic_models,
|
2178 |
**german_models,
|
2179 |
**french_models,
|
|
|
2180 |
**russian_models,
|
2181 |
**korean_models,
|
2182 |
**thai_models,
|
@@ -2202,6 +2218,7 @@ language_to_models = {
|
|
2202 |
"Korean": list(korean_models.keys()),
|
2203 |
"Portuguese (Brazil)": list(portuguese_brazlian_models.keys()),
|
2204 |
"Russian": list(russian_models.keys()),
|
|
|
2205 |
"Thai": list(thai_models.keys()),
|
2206 |
"Tibetan": list(tibetan_models.keys()),
|
2207 |
"Vietnamese": list(vietnamese_models.keys()),
|
|
|
214 |
return french_models[repo_id](
|
215 |
repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
|
216 |
)
|
217 |
+
elif repo_id in spanish_models:
|
218 |
+
return spanish_models[repo_id](
|
219 |
+
repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
|
220 |
+
)
|
221 |
elif repo_id in japanese_models:
|
222 |
return japanese_models[repo_id](
|
223 |
repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
|
|
|
1591 |
"csukuangfj/sherpa-onnx-streaming-zipformer-bilingual-zh-en-2023-02-20",
|
1592 |
"k2-fsa/sherpa-onnx-streaming-zipformer-korean-2024-06-16",
|
1593 |
"csukuangfj/sherpa-onnx-streaming-zipformer-en-kroko-2025-08-06",
|
1594 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-es-kroko-2025-08-06",
|
1595 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-fr-kroko-2025-08-06",
|
1596 |
], repo_id
|
1597 |
|
1598 |
+
if repo_id in [
|
1599 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-en-kroko-2025-08-06",
|
1600 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-es-kroko-2025-08-06",
|
1601 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-fr-kroko-2025-08-06",
|
1602 |
+
]:
|
1603 |
encoder_model = _get_nn_model_filename(
|
1604 |
repo_id=repo_id,
|
1605 |
filename="encoder.onnx",
|
|
|
2122 |
|
2123 |
french_models = {
|
2124 |
"shaojieli/sherpa-onnx-streaming-zipformer-fr-2023-04-14": _get_french_pre_trained_model,
|
2125 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-fr-kroko-2025-08-06": _get_streaming_zipformer_pre_trained_model,
|
2126 |
+
}
|
2127 |
+
|
2128 |
+
spanish_models = {
|
2129 |
+
"csukuangfj/sherpa-onnx-streaming-zipformer-es-kroko-2025-08-06": _get_streaming_zipformer_pre_trained_model,
|
2130 |
}
|
2131 |
|
2132 |
japanese_models = {
|
|
|
2192 |
**arabic_models,
|
2193 |
**german_models,
|
2194 |
**french_models,
|
2195 |
+
**spanish_models,
|
2196 |
**russian_models,
|
2197 |
**korean_models,
|
2198 |
**thai_models,
|
|
|
2218 |
"Korean": list(korean_models.keys()),
|
2219 |
"Portuguese (Brazil)": list(portuguese_brazlian_models.keys()),
|
2220 |
"Russian": list(russian_models.keys()),
|
2221 |
+
"Spanish": list(spanish_models.keys()),
|
2222 |
"Thai": list(thai_models.keys()),
|
2223 |
"Tibetan": list(tibetan_models.keys()),
|
2224 |
"Vietnamese": list(vietnamese_models.keys()),
|