101Frost commited on
Commit
c64435a
·
verified ·
1 Parent(s): 1acef58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -19,10 +19,11 @@ MODELS = {
19
  "epitran": epitran.Epitran("ara-Arab")
20
  },
21
  "English": {
22
- "processor": Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h"),
23
- "model": Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h"),
24
- "epitran": None # Not needed; using eng_to_ipa
25
  }
 
26
  }
27
 
28
  # Suppress the warning about newly initialized weights
 
19
  "epitran": epitran.Epitran("ara-Arab")
20
  },
21
  "English": {
22
+ "processor": Wav2Vec2Processor.from_pretrained("jonatasgrosman/wav2vec2-large-xlsr-53-english"),
23
+ "model": Wav2Vec2ForCTC.from_pretrained("jonatasgrosman/wav2vec2-large-xlsr-53-english"),
24
+ "epitran": epitran.Epitran("eng-Latn")
25
  }
26
+
27
  }
28
 
29
  # Suppress the warning about newly initialized weights