Commit
路
c64cc74
1
Parent(s):
046b16f
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,10 +8,10 @@ from transformers import SpeechT5ForTextToSpeech, SpeechT5HifiGan, SpeechT5Proce
|
|
| 8 |
|
| 9 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 10 |
|
| 11 |
-
#
|
| 12 |
asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)
|
| 13 |
|
| 14 |
-
#
|
| 15 |
processor = SpeechT5Processor.from_pretrained("jjsprockel/speecht5_finetuned_voxpopuli_it")
|
| 16 |
|
| 17 |
model = SpeechT5ForTextToSpeech.from_pretrained("jjsprockel/speecht5_finetuned_voxpopuli_it").to(device)
|
|
@@ -39,7 +39,7 @@ def speech_to_speech_translation(audio):
|
|
| 39 |
return 16000, synthesised_speech
|
| 40 |
|
| 41 |
|
| 42 |
-
title = "
|
| 43 |
description = """
|
| 44 |
Demostraci贸n de un traductor de voz a voz en cascada (cascaded speech-to-speech translation o STST), funciona mapeando de una fuente de voz en cualquier lenguaje a una meta de texto en Italiano. Se usa el modelo de OpenAI [Whisper Base](https://huggingface.co/openai/whisper-base) para la traducci贸n, y el modelo de Microsoft
|
| 45 |
[SpeechT5 TTS](https://huggingface.co/microsoft/speecht5_tts) sometido a un ajuse fino para pasar de texto-a-voz:
|
|
|
|
| 8 |
|
| 9 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 10 |
|
| 11 |
+
# carguemos el checkpoint para translaci贸n del habla
|
| 12 |
asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)
|
| 13 |
|
| 14 |
+
# carguemos el checkpoint texto-a-habla (text-to-speech) as铆 como el speaker embeddings
|
| 15 |
processor = SpeechT5Processor.from_pretrained("jjsprockel/speecht5_finetuned_voxpopuli_it")
|
| 16 |
|
| 17 |
model = SpeechT5ForTextToSpeech.from_pretrained("jjsprockel/speecht5_finetuned_voxpopuli_it").to(device)
|
|
|
|
| 39 |
return 16000, synthesised_speech
|
| 40 |
|
| 41 |
|
| 42 |
+
title = "Traductor al Italiano de Voz a Voz en Cascada"
|
| 43 |
description = """
|
| 44 |
Demostraci贸n de un traductor de voz a voz en cascada (cascaded speech-to-speech translation o STST), funciona mapeando de una fuente de voz en cualquier lenguaje a una meta de texto en Italiano. Se usa el modelo de OpenAI [Whisper Base](https://huggingface.co/openai/whisper-base) para la traducci贸n, y el modelo de Microsoft
|
| 45 |
[SpeechT5 TTS](https://huggingface.co/microsoft/speecht5_tts) sometido a un ajuse fino para pasar de texto-a-voz:
|