anyantudre commited on
Commit
df4c0ad
·
verified ·
1 Parent(s): db3d44a

Update src/speech_to_text.py

Browse files
Files changed (1) hide show
  1. src/speech_to_text.py +1 -1
src/speech_to_text.py CHANGED
@@ -27,7 +27,7 @@ def transcribe(fp:str, target_lang:str) -> str:
27
  start_time = time.time()
28
 
29
  # Load transcription model
30
- model_id = "facebook/mms-1b-all"
31
 
32
  processor = AutoProcessor.from_pretrained(model_id, target_lang=target_lang)
33
  model = Wav2Vec2ForCTC.from_pretrained(model_id, target_lang=target_lang, ignore_mismatched_sizes=True)
 
27
  start_time = time.time()
28
 
29
  # Load transcription model
30
+ model_id = "anyantudre/wav2vec2-large-mms-1b-mos-V1"
31
 
32
  processor = AutoProcessor.from_pretrained(model_id, target_lang=target_lang)
33
  model = Wav2Vec2ForCTC.from_pretrained(model_id, target_lang=target_lang, ignore_mismatched_sizes=True)