Update model.py

model.py

@@ -371,36 +371,44 @@ def _get_vietnamese_pretrained_model(
         "csukuangfj/sherpa-onnx-zipformer-vi-2025-04-20",
     ), repo_id
 
-    decoder_model = _get_nn_model_filename(
-        repo_id=repo_id,
-        filename="decoder-epoch-12-avg-8.onnx",
-        subfolder=".",
-    )
+    # decoder_model = _get_nn_model_filename(
+    #     repo_id=repo_id,
+    #     filename="decoder-epoch-12-avg-8.onnx",
+    #     subfolder=".",
+    # )
 
-    if repo_id == "csukuangfj/sherpa-onnx-zipformer-vi-int8-2025-04-20":
-        encoder_model = _get_nn_model_filename(
-            repo_id=repo_id,
-            filename="encoder-epoch-12-avg-8.int8.onnx",
-            subfolder=".",
-        )
+    decoder_model = "decoder-epoch-45-avg-25.onnx"
 
-        joiner_model = _get_nn_model_filename(
-            repo_id=repo_id,
-            filename="joiner-epoch-12-avg-8.int8.onnx",
-            subfolder=".",
-        )
+    if repo_id == "csukuangfj/sherpa-onnx-zipformer-vi-int8-2025-04-20":
+        # encoder_model = _get_nn_model_filename(
+        #     repo_id=repo_id,
+        #     filename="encoder-epoch-12-avg-8.int8.onnx",
+        #     subfolder=".",
+        # )
+
+        encoder_model = "encoder-epoch-45-avg-25.int8.onnx"
+
+        # joiner_model = _get_nn_model_filename(
+        #     repo_id=repo_id,
+        #     filename="joiner-epoch-12-avg-8.int8.onnx",
+        #     subfolder=".",
+        # )
+        joiner_model = "joiner-epoch-45-avg-25.int8.onnx"
     elif repo_id == "csukuangfj/sherpa-onnx-zipformer-vi-2025-04-20":
-        encoder_model = _get_nn_model_filename(
-            repo_id=repo_id,
-            filename="encoder-epoch-12-avg-8.onnx",
-            subfolder=".",
-        )
-
-        joiner_model = _get_nn_model_filename(
-            repo_id=repo_id,
-            filename="joiner-epoch-12-avg-8.onnx",
-            subfolder=".",
-        )
+        # encoder_model = _get_nn_model_filename(
+        #     repo_id=repo_id,
+        #     filename="encoder-epoch-12-avg-8.onnx",
+        #     subfolder=".",
+        # )
+
+        encoder_model = "encoder-epoch-45-avg-25.onnx"
+
+        # joiner_model = _get_nn_model_filename(
+        #     repo_id=repo_id,
+        #     filename="joiner-epoch-12-avg-8.onnx",
+        #     subfolder=".",
+        # )
+        joiner_model = "joiner-epoch-45-avg-25.onnx"
     else:
         raise ValueError(f"repo_id: {repo_id}")