Bismillah

.gitignore

@@ -0,0 +1 @@
+.idea

README.md

@@ -1,8 +1,8 @@
 ---
-title: Latest Demo
-emoji: 🐨
-colorFrom: pink
-colorTo: green
+title: Tarteel AI's - Latest Model Demo
+emoji: 🤗
+colorFrom: green
+colorTo: blue
 sdk: gradio
 sdk_version: 3.0.5
 app_file: app.py

app.py

@@ -0,0 +1,36 @@
+import gradio as gr
+from gradio import mix
+from gradio import inputs, outputs  # noqa: F401
+import hydra
+from omegaconf import DictConfig
+
+from models_evaluation.models import BaseModel
+
+
+@hydra.main(version_base=None, config_path="conf", config_name="gradio-asr.yaml")
+def main(cfg: DictConfig):
+    audio_input = gr.inputs.Audio(source=cfg.gradio.source, type="file", label="Audio")
+
+    default_kwargs = {
+        "theme": "dark-huggingface",
+        "allow_flagging": "manual",
+        "title": cfg.gradio.title,
+    }
+
+    all_io = []
+    models = []
+    for model_cfg in cfg.models:
+        model = hydra.utils.instantiate(model_cfg)  # type: BaseModel
+        model_cfg = model_cfg.cfg
+        models.append(model)
+        io = gr.Interface(
+            model.transcribe, inputs=audio_input, outputs=gr.outputs.JSON(label=model_cfg.name)
+        )
+        all_io.append(io)
+
+    parallel = mix.Parallel(*all_io, **default_kwargs)
+    parallel.launch(share=True, debug=True)
+
+
+if __name__ == "__main__":
+    main()

conf/gradio-asr.yaml

@@ -0,0 +1,28 @@
+sample_rate: &sample_rate 16000
+device: &device "cuda"
+
+gradio:
+  source: "upload"  # 'upload' or 'microphone'
+  title: "CitriNet(Lg) Un/Diacritized Comparison"
+
+models:
+  - _target_: "models_evaluation.models.citrinet.CitrinetTorch"
+    cfg:
+      model_target: "nemo.collections.asr.models.EncDecCTCModelBPE"
+      name: "CnLg_SpeUni1024_DI_EATL600"
+      sample_rate: *sample_rate
+      device: *device
+      model_path: "wandb://tarteel/nemo-experiments/CnLg_SpeUni1024_DI_EATL600:v1"
+
+  - _target_: "models_evaluation.models.citrinet.CitrinetTorch"
+    cfg:
+      model_target: "nemo.collections.asr.models.EncDecCTCModelBPE"
+      name: "CnLg-SpeUni256-EATL1300"
+      sample_rate: *sample_rate
+      device: *device
+      model_path: "wandb://tarteel/nemo-experiments/CnLg-SpeUni256-EATL1300:v1"
+
+
+hydra:
+  run:
+    dir: ..

requirements.txt

@@ -0,0 +1,4 @@
+nemo_toolkit[asr]==1.7.2
+hydra-core
+wandb
+git+https://[email protected]/Tarteel-io/models_evaluation.git@main#egg=models_evaluation[nemo]