Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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/
|
| 58 |
-
|
| 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()
|