LT4Ryan commited on
Commit
67c888d
·
verified ·
1 Parent(s): 14ff637

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,7 +12,8 @@ import gradio.themes as gr_themes
12
  import csv
13
 
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
- MODEL_NAME="nvidia/parakeet-tdt-0.6b-v2"
 
16
 
17
  model = ASRModel.from_pretrained(model_name=MODEL_NAME)
18
  model.eval()
 
12
  import csv
13
 
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
+ MODEL_NAME="nvidia/parakeet-rnnt-1.1b"
16
+ #"nvidia/parakeet-tdt-0.6b-v2"
17
 
18
  model = ASRModel.from_pretrained(model_name=MODEL_NAME)
19
  model.eval()