jjsprockel commited on
Commit
c64cc74
1 Parent(s): 046b16f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- # load speech translation checkpoint
12
  asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)
13
 
14
- # load text-to-speech checkpoint and 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,7 +39,7 @@ def speech_to_speech_translation(audio):
39
  return 16000, synthesised_speech
40
 
41
 
42
- title = "Cascaded STST"
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: