bel32123 commited on
Commit
eeb7735
·
1 Parent(s): fa4bb60

Fix path to kenlm model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def load_model():
11
  path = os.path.join(os.getcwd(), "wav2vecasr", "model", "checkpoint-600")
12
  asr_model = Wav2Vec2OptimisedPhonemeASRModel(path, os.path.join(path, "wav2vec2_vocab_final.json"),
13
  os.path.join(os.getcwd(), "wav2vecasr", "pretrained_models",
14
- "ken-lm-ngram"))
15
  g2p = GraphemeToPhoneme.from_hparams("speechbrain/soundchoice-g2p")
16
  mispronounciation_detector = MispronounciationDetector(asr_model, g2p, "cpu")
17
  return mispronounciation_detector
 
11
  path = os.path.join(os.getcwd(), "wav2vecasr", "model", "checkpoint-600")
12
  asr_model = Wav2Vec2OptimisedPhonemeASRModel(path, os.path.join(path, "wav2vec2_vocab_final.json"),
13
  os.path.join(os.getcwd(), "wav2vecasr", "pretrained_models",
14
+ "en-kenlm-model", "en.arpa.bin"))
15
  g2p = GraphemeToPhoneme.from_hparams("speechbrain/soundchoice-g2p")
16
  mispronounciation_detector = MispronounciationDetector(asr_model, g2p, "cpu")
17
  return mispronounciation_detector