fix: run nemo_asr task for mbaza_ctc_large model

nikochiko · nikochiko · commit 7f95f9e08169 · 2025-08-25T17:25:31.000+05:30
diff --git a/daras_ai_v2/asr.py b/daras_ai_v2/asr.py
@@ -1218,7 +1218,7 @@ def run_asr(
             for result in batch.transcript.results  # SpeechRecognitionResult
             if result.alternatives
         )
-    elif "nemo" in selected_model.name:
+    elif selected_model == AsrModels.mbaza_ctc_large:
         data = call_celery_task(
             "nemo_asr",
             pipeline=dict(

Original file line number	Diff line number	Diff line change
`@@ -1218,7 +1218,7 @@ def run_asr(`
`1218`	`1218`	`for result in batch.transcript.results # SpeechRecognitionResult`
`1219`	`1219`	`if result.alternatives`
`1220`	`1220`	`)`
`1221`		`- elif "nemo" in selected_model.name:`
	`1221`	`+ elif selected_model == AsrModels.mbaza_ctc_large:`
`1222`	`1222`	`data = call_celery_task(`
`1223`	`1223`	`"nemo_asr",`
`1224`	`1224`	`pipeline=dict(`