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()),
|