DrishtiSharma commited on
Commit
4e336ff
1 Parent(s): edbc412

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def load_model(model_path="facebook/wav2vec2-large-robust-ft-swbd-300h"):
28
  model = AutoModelForCTC.from_pretrained(model_path).to(device)
29
  return processor, model
30
 
31
- model_path = st.radio("Select a model", ("jonatasgrosman/wav2vec2-large-xlsr-53-spanish", "patrickvonplaten/wav2vec2-large-xlsr-53-spanish-with-lm", "jonatasgrosman/wav2vec2-xls-r-1b-spanish", "facebook/wav2vec2-large-xlsr-53-spanish", "glob-asr/xls-r-es-test-lm"))
32
 
33
  processor, model = load_model(model_path)
34
 
 
28
  model = AutoModelForCTC.from_pretrained(model_path).to(device)
29
  return processor, model
30
 
31
+ model_path = st.radio("Select a model", ("jonatasgrosman/wav2vec2-xls-r-1b-spanish", "jonatasgrosman/wav2vec2-large-xlsr-53-spanish", "patrickvonplaten/wav2vec2-large-xlsr-53-spanish-with-lm", "facebook/wav2vec2-large-xlsr-53-spanish", "glob-asr/xls-r-es-test-lm"))
32
 
33
  processor, model = load_model(model_path)
34