DrishtiSharma
commited on
Commit
•
4e336ff
1
Parent(s):
edbc412
Update app.py
Browse files
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-
|
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 |
|