From ee1631e8a8c028ad09a3a60056e6820b69eb792b Mon Sep 17 00:00:00 2001
From: Somshubra Majumdar <titu1994@gmail.com>
Date: Mon, 2 May 2022 01:40:33 -0700
Subject: [PATCH] Update NeMo ASR resolver (#126)

* Update ASR resolver

* Correct method
---
 js/src/lib/interfaces/Libraries.ts | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/js/src/lib/interfaces/Libraries.ts b/js/src/lib/interfaces/Libraries.ts
index 168aa9938..5d5d65829 100644
--- a/js/src/lib/interfaces/Libraries.ts
+++ b/js/src/lib/interfaces/Libraries.ts
@@ -321,10 +321,7 @@ const nemoDomainResolver = (domain: string, model: ModelData): string | undefine
 	switch (domain) {
 		case "ASR":
 			return `import nemo.collections.asr as nemo_asr
-from huggingface_hub import hf_hub_download
-
-path = hf_hub_download(repo_id="${model.id}", filename="${modelName}")
-asr_model = nemo_asr.models.ASRModel.restore_from(path)
+asr_model = nemo_asr.models.ASRModel.from_pretrained("${model.id}")
 
 transcriptions = asr_model.transcribe(["file.wav"])`;
 		default: