ayymen commited on
Commit
f0a2f89
·
verified ·
1 Parent(s): 5be4c70

Use Kabyle model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import gradio as gr
7
  from pydub import AudioSegment
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
- MODEL_NAME="nvidia/parakeet-rnnt-1.1b"
11
  YT_LENGTH_LIMIT_S=3600
12
 
13
  model = EncDecRNNTBPEModel.from_pretrained(model_name=MODEL_NAME).to(device)
 
7
  from pydub import AudioSegment
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
+ MODEL_NAME="nvidia/stt_kab_conformer_transducer_large"
11
  YT_LENGTH_LIMIT_S=3600
12
 
13
  model = EncDecRNNTBPEModel.from_pretrained(model_name=MODEL_NAME).to(device)