Spaces:
Sleeping
Sleeping
anyantudre
commited on
Update src/speech_to_text.py
Browse files- 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 = "
|
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)
|