msis commited on
Commit
4087c59
1 Parent(s): c34f6a6

Update model restoration method

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def lazy_load_models(models_names):
40
  for model_name in models_names:
41
  model = models[model_name]
42
  if not model:
43
- models[model_name] = nemo_asr.models.ASRModel.restore_from(
44
  list(filter(lambda x: x.name == model_name, latest_artifacts))[0].file()
45
  )
46
  models[model_name].eval()
 
40
  for model_name in models_names:
41
  model = models[model_name]
42
  if not model:
43
+ models[model_name] = nemo_asr.models.EncDecCTCModelBPE.restore_from(
44
  list(filter(lambda x: x.name == model_name, latest_artifacts))[0].file()
45
  )
46
  models[model_name].eval()