Fix decoder vocab size

README.md

@@ -1,10 +1,9 @@
+
 ---
-language:
+language: 
 - eu
+
 license: apache-2.0
-base_model:
-- HiTZ/stt_eu_conformer_transducer_large
-pipeline_tag: automatic-speech-recognition
 ---
 
 # OVOS - STT Conformer Transducer Large ONNX (Basque)
@@ -44,6 +43,10 @@ model = nemo_asr.models.ASRModel.from_pretrained("HiTZ/stt_eu_conformer_transduc
 
 onnx_dir = Path("onnx-dir")
 onnx_dir.mkdir(exist_ok=True)
+
+if hasattr(model, "decoder") and hasattr(model.decoder, "vocab_size"):
+    model.decoder.vocab_size += 1  # to account for <blk>
+
 model.export(str(Path(onnx_dir, "model.onnx")))
 
 with Path(onnx_dir, "vocab.txt").open("wt") as f:
@@ -53,4 +56,5 @@ with Path(onnx_dir, "vocab.txt").open("wt") as f:
 
 # Licensing
 
-The license is derived from the original model: Apache 2.0. For more details, please refer to [HiTZ/stt_eu_conformer_transducer_large](https://huggingface.co/HiTZ/stt_eu_conformer_transducer_large).
+The license is derived from the original model: Apache 2.0. For more details, please refer to [HiTZ/stt_eu_conformer_transducer_large](https://huggingface.co/HiTZ/stt_eu_conformer_transducer_large).
+