AshwinSankar commited on
Commit
f990271
·
verified ·
1 Parent(s): 40c2c9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -54,9 +54,9 @@ This Gradio demo showcases **IndicConformer**, a speech recognition model for **
54
  hf_token = os.getenv("HF_TOKEN")
55
  device = "cuda:0" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
56
  torch_dtype = torch.bfloat16 if device != "cpu" else torch.float32
57
- model_name_or_path = "ai4bharat/indicconformer_stt_bn_hybrid_ctc_rnnt_large"
58
- # model = nemo_asr.models.EncDecCTCModel.from_pretrained(model_name_or_path).to(device)
59
- model = nemo_asr.models.EncDecCTCModel.restore_from("indicconformer_stt_bn_hybrid_rnnt_large.nemo").to(device)
60
  model.eval()
61
 
62
  CACHE_EXAMPLES = os.getenv("CACHE_EXAMPLES") == "1" and torch.cuda.is_available()
 
54
  hf_token = os.getenv("HF_TOKEN")
55
  device = "cuda:0" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
56
  torch_dtype = torch.bfloat16 if device != "cpu" else torch.float32
57
+ model_name_or_path = "ai4bharat/IndicConformer"
58
+ model = nemo_asr.models.EncDecCTCModel.from_pretrained(model_name_or_path).to(device)
59
+ # model = nemo_asr.models.EncDecCTCModel.restore_from("indicconformer_stt_bn_hybrid_rnnt_large.nemo").to(device)
60
  model.eval()
61
 
62
  CACHE_EXAMPLES = os.getenv("CACHE_EXAMPLES") == "1" and torch.cuda.is_available()