Adding pretrained French ASR models to ctc_bpe and rnnt_bpe listings (#…

…3225) * Pushing WFST_tutorial for open draft. (Still need to review collab code. Signed-off-by: tbartley94 <tbartley@nvidia.com> * Checked tutorial code for WFST_Tutorial is properly functioning. Also included some formatting edits. Signed-off-by: tbartley94 <tbartley@nvidia.com> * Responding to editorial comments for WFST_tutorial Signed-off-by: tbartley94 <tbartley@nvidia.com> * Added images to folder and wrote README for tutorials Signed-off-by: tbartley94 <tbartley@nvidia.com> * Few more editorial changes to explain permutations in classification. Signed-off-by: tbartley94 <tbartley@nvidia.com> * Updated tutorials documentation page. Signed-off-by: tbartley94 <tbartley@nvidia.com> * Forgot links for README Signed-off-by: tbartley94 <tbartley@nvidia.com> * TOC links were dead Signed-off-by: tbartley94 <tbartley@nvidia.com> * More dead links to fix. Signed-off-by: tbartley94 <tbartley@nvidia.com> * removing collab install and appending a warning instead. Signed-off-by: tbartley94 <tbartley@nvidia.com> * Update WFST_Tutorial.ipynb Signed-off-by: tbartley94 <tbartley@nvidia.com> * Adding pretrained French models to ctc_bpe_models and rnnt_bpe_models available models listing Signed-off-by: tbartley94 <tbartley@nvidia.com> Co-authored-by: Oleksii Kuchaiev <okuchaiev@users.noreply.github.com> Co-authored-by: Yang Zhang <yzhang123@users.noreply.github.com> Signed-off-by: Virginia Adams <vadams@nvidia.com>
NVIDIA · Dec 9, 2021 · 4673076 · 4673076
1 parent 258f473
commit 4673076
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 3 deletions.
diff --git a/nemo/collections/asr/models/ctc_bpe_models.py b/nemo/collections/asr/models/ctc_bpe_models.py
@@ -101,7 +101,20 @@ def list_available_models(cls) -> Optional[PretrainedModelInfo]:
             description="For details about this model, please visit https://ngc.nvidia.com/catalog/models/nvidia:nemo:stt_de_citrinet_1024",
             location="https://api.ngc.nvidia.com/v2/models/nvidia/nemo/stt_de_citrinet_1024/versions/1.3.2/files/stt_de_citrinet_1024.nemo",
         )
+        results.append(model)
 
+        model = PretrainedModelInfo(
+            pretrained_model_name="stt_fr_citrinet_1024_gamma_0_25",
+            description="For details about this model, please visit https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/stt_fr_citrinet_1024_gamma_0_25",
+            location="https://api.ngc.nvidia.com/v2/models/nvidia/nemo/stt_fr_citrinet_1024_gamma_0_25/versions/1.5/files/stt_fr_citrinet_1024_gamma_0_25.nemo",
+        )
+        results.append(model)
+
+        model = PretrainedModelInfo(
+            pretrained_model_name="stt_fr_no_hyphen_citrinet_1024_gamma_0_25",
+            description="For details about this model, please visit https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/stt_fr_citrinet_1024_gamma_0_25",
+            location="https://api.ngc.nvidia.com/v2/models/nvidia/nemo/stt_fr_citrinet_1024_gamma_0_25/versions/1.5/files/stt_fr_no_hyphen_citrinet_1024_gamma_0_25.nemo",
+        )
         results.append(model)
 
         model = PretrainedModelInfo(
@@ -146,6 +159,13 @@ def list_available_models(cls) -> Optional[PretrainedModelInfo]:
         )
         results.append(model)
 
+        model = PretrainedModelInfo(
+            pretrained_model_name="stt_fr_conformer_ctc_large",
+            description="For details about this model, please visit https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/stt_fr_conformer_ctc_large",
+            location="https://api.ngc.nvidia.com/v2/models/nvidia/nemo/stt_fr_conformer_ctc_large/versions/1.5/files/stt_fr_conformer_ctc_large.nemo",
+        )
+        results.append(model)
+
         return results
 
     def __init__(self, cfg: DictConfig, trainer=None):

diff --git a/nemo/collections/asr/models/rnnt_bpe_models.py b/nemo/collections/asr/models/rnnt_bpe_models.py
@@ -115,9 +115,9 @@ def list_available_models(cls) -> List[PretrainedModelInfo]:
         results.append(model)
 
         model = PretrainedModelInfo(
-            pretrained_model_name="stt_de_contextnet_1024",
-            description="For details about this model, please visit https://ngc.nvidia.com/catalog/models/nvidia:nemo:stt_de_contextnet_1024",
-            location="https://api.ngc.nvidia.com/v2/models/nvidia/nemo/stt_de_contextnet_1024/versions/1.4.0/files/stt_de_contextnet_1024.nemo",
+            pretrained_model_name="stt_fr_contextnet_1024",
+            description="For details about this model, please visit https://ngc.nvidia.com/catalog/models/nvidia:nemo:stt_fr_contextnet_1024",
+            location="https://api.ngc.nvidia.com/v2/models/nvidia/nemo/stt_fr_contextnet_1024/versions/1.5/files/stt_fr_contextnet_1024.nemo",
         )
         results.append(model)