Hugging Face's logo

Spaces:
preetam8
/
speech-to-speech-translation
Running

App Files Community
preetam8 commited on
Commit
594d8c7
·
1 Parent(s): 8cf6151

use whisper medium

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ target_language = "fr"
14
 
15
  # load speech translation checkpoint
16
  # asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)
17
- whisper_model_name = "openai/whisper-small"
18
  whisper_processor = WhisperProcessor.from_pretrained(whisper_model_name)
19
  whisper_model = WhisperForConditionalGeneration.from_pretrained(whisper_model_name)
20
  decoder_ids = whisper_processor.get_decoder_prompt_ids(language=target_language, task="transcribe")
 
14
 
15
  # load speech translation checkpoint
16
  # asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)
17
+ whisper_model_name = "openai/whisper-medium"
18
  whisper_processor = WhisperProcessor.from_pretrained(whisper_model_name)
19
  whisper_model = WhisperForConditionalGeneration.from_pretrained(whisper_model_name)
20
  decoder_ids = whisper_processor.get_decoder_prompt_ids(language=target_language, task="transcribe")