Update app.py
Browse files
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-
|
|
|
|
| 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()
|