mizoru commited on
Commit
04026a3
1 Parent(s): d9891ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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 = sorted(MODELS.keys())
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