Spaces:
Sleeping
Sleeping
Update app.py
Browse filesChange from Whisper base to Whisper small
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-
|
9 |
pipe = pipeline("automatic-speech-recognition",
|
10 |
-
model="openai/whisper-
|
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
|
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 |
"""
|