Hugging Face's logo

Spaces:
harshitv804
/
Tamil_Translator
Runtime error

App Files Community
harshitv804 commited on
Commit
4c24256
·
1 Parent(s): 2cb4c4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import WhisperProcessor, WhisperForConditionalGeneration
3
  import torchaudio
4
 
5
  def translate(audio):
6
- model_id_asr = "openai/whisper-medium"
7
  processor_asr = WhisperProcessor.from_pretrained(model_id_asr)
8
  model_asr = WhisperForConditionalGeneration.from_pretrained(model_id_asr)
9
  forced_decoder_ids = processor_asr.get_decoder_prompt_ids(language="tamil", task="translate")
 
3
  import torchaudio
4
 
5
  def translate(audio):
6
+ model_id_asr = "openai/whisper-small"
7
  processor_asr = WhisperProcessor.from_pretrained(model_id_asr)
8
  model_asr = WhisperForConditionalGeneration.from_pretrained(model_id_asr)
9
  forced_decoder_ids = processor_asr.get_decoder_prompt_ids(language="tamil", task="translate")