camanalo1 commited on
Commit
e7cab25
1 Parent(s): 0c0df10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,10 +1,13 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
 
 
 
4
  # Define the pipeline for speech recognition with the NeMo Canary model
5
  pipe = pipeline(
6
  task="automatic-speech-recognition",
7
- model="nvidia/canary-1b"
8
  )
9
 
10
  # Create a Gradio interface from the pipeline
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Define the full path to the NeMo Canary model
5
+ model_name = "nvidia/canary-1b"
6
+
7
  # Define the pipeline for speech recognition with the NeMo Canary model
8
  pipe = pipeline(
9
  task="automatic-speech-recognition",
10
+ model=model_name
11
  )
12
 
13
  # Create a Gradio interface from the pipeline