Ellight commited on
Commit
9ed3d6d
·
verified ·
1 Parent(s): b9bb0a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import SpeechT5ForTextToSpeech, SpeechT5HifiGan, SpeechT5Proce
8
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
 
10
  # load speech translation checkpoint
11
- asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-large-v2", device=device)
12
 
13
  # load text-to-speech checkpoint and speaker embeddings
14
  model_id = "Ellight/speecht5_finetuned_voxpopuli_nl" # update with your model id
 
8
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
 
10
  # load speech translation checkpoint
11
+ asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-tiny", device=device)
12
 
13
  # load text-to-speech checkpoint and speaker embeddings
14
  model_id = "Ellight/speecht5_finetuned_voxpopuli_nl" # update with your model id