sample_rate: &sample_rate 16000
device: &device "cpu"

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: ..