Spaces:

navinaananthan
/

Sinhala-Speech-Translation

Runtime error

navinaananthan commited on Sep 12, 2023

Commit

82d7125

1 Parent(s): 1ef3084

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,8 +34,8 @@ model = T5ForConditionalGeneration.from_pretrained(model_name)
 '''
 Importing necessary models for ASR module
 '''
-asr_processor = WhisperProcessor.from_pretrained("whisper-small-si_50k_25jul23_checkpoint-600")
-asr_model = WhisperForConditionalGeneration.from_pretrained("whisper-small-si_50k_25jul23_checkpoint-600")
 forced_decoder_ids = asr_processor.get_decoder_prompt_ids(language="sinhala", task="transcribe")

 '''
 Importing necessary models for ASR module
 '''
+asr_processor = WhisperProcessor.from_pretrained("openai/whisper-large-v2")
+asr_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large-v2")
 forced_decoder_ids = asr_processor.get_decoder_prompt_ids(language="sinhala", task="transcribe")