Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -3,12 +3,14 @@ from transformers import pipeline, AutoModelForCTC, Wav2Vec2Processor, Wav2Vec2P
|
|
3 |
|
4 |
MODELS = {
|
5 |
"Tatar": {"model_id": "sammy786/wav2vec2-xlsr-tatar", "has_lm": False},
|
6 |
-
"Chuvash": {"model_id": "sammy786/wav2vec2-xlsr-chuvash", "has_lm": False}
|
|
|
|
|
7 |
}
|
8 |
|
9 |
CACHED_MODELS_BY_ID = {}
|
10 |
|
11 |
-
LANGUAGES =
|
12 |
|
13 |
def run(input_file, language, decoding_type):
|
14 |
|
|
|
3 |
|
4 |
MODELS = {
|
5 |
"Tatar": {"model_id": "sammy786/wav2vec2-xlsr-tatar", "has_lm": False},
|
6 |
+
"Chuvash": {"model_id": "sammy786/wav2vec2-xlsr-chuvash", "has_lm": False},
|
7 |
+
"Bashkir": {"model_id": "AigizK/wav2vec2-large-xls-r-300m-bashkir-cv7_opt", "has_lm": True},
|
8 |
+
"Erzya": {"model_id": "DrishtiSharma/wav2vec2-large-xls-r-300m-myv-v1", "has_lm": True}
|
9 |
}
|
10 |
|
11 |
CACHED_MODELS_BY_ID = {}
|
12 |
|
13 |
+
LANGUAGES = (MODELS.keys())
|
14 |
|
15 |
def run(input_file, language, decoding_type):
|
16 |
|