camanalo1 commited on
Commit
d828f00
·
verified ·
1 Parent(s): 7814090

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import torch
3
  from transformers import pipeline
 
4
 
5
  # Load Canary ASR model
6
  canary_model = EncDecMultiTaskModel.from_pretrained('nvidia/canary-1b')
 
1
  import gradio as gr
2
  import torch
3
  from transformers import pipeline
4
+ from nemo.collections.asr.models import EncDecMultiTaskModel # Add this import statement
5
 
6
  # Load Canary ASR model
7
  canary_model = EncDecMultiTaskModel.from_pretrained('nvidia/canary-1b')