64FC commited on
Commit
20945f0
1 Parent(s): faa1ad3

Update app.py

Browse files

Change from Whisper base to Whisper small

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,9 +5,9 @@ import gradio as gr
5
 
6
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
7
 
8
- # Load Whisper-base
9
  pipe = pipeline("automatic-speech-recognition",
10
- model="openai/whisper-base",
11
  device=device
12
  )
13
 
@@ -46,7 +46,7 @@ def speech_to_speech_translation(audio):
46
  # Define the title etc
47
  title = "Cascaded STST"
48
  description = """
49
- Demo for cascaded speech-to-speech translation (STST), mapping from source speech in any language to target speech in French. Demo uses OpenAI's [Whisper Base](https://huggingface.co/openai/whisper-base) model for speech translation, and Facebook's
50
  [MMS TTS](https://huggingface.co/facebook/mms-tts) model, finetuned by [Matthijs](https://huggingface.co/Matthijs), for text-to-speech:
51
  ![Cascaded STST](https://huggingface.co/datasets/huggingface-course/audio-course-images/resolve/main/s2st_cascaded.png "Diagram of cascaded speech to speech translation")
52
  """
 
5
 
6
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
7
 
8
+ # Load Whisper-small
9
  pipe = pipeline("automatic-speech-recognition",
10
+ model="openai/whisper-small",
11
  device=device
12
  )
13
 
 
46
  # Define the title etc
47
  title = "Cascaded STST"
48
  description = """
49
+ Demo for cascaded speech-to-speech translation (STST), mapping from source speech in any language to target speech in French. Demo uses OpenAI's [Whisper Small](https://huggingface.co/openai/whisper-small) model for speech translation, and Facebook's
50
  [MMS TTS](https://huggingface.co/facebook/mms-tts) model, finetuned by [Matthijs](https://huggingface.co/Matthijs), for text-to-speech:
51
  ![Cascaded STST](https://huggingface.co/datasets/huggingface-course/audio-course-images/resolve/main/s2st_cascaded.png "Diagram of cascaded speech to speech translation")
52
  """