Spaces:
Runtime error
Runtime error
Upload app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ from transformers import SpeechT5ForTextToSpeech, SpeechT5HifiGan, SpeechT5Proce
|
|
22 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
23 |
|
24 |
# load speech translation checkpoint
|
25 |
-
asr_pipe = pipeline("automatic-speech-recognition", model="
|
26 |
|
27 |
# !pip -q install sentencepiece
|
28 |
# load text-to-speech checkpoint and speaker embeddings
|
|
|
22 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
23 |
|
24 |
# load speech translation checkpoint
|
25 |
+
asr_pipe = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-xls-r-300m", device=device)
|
26 |
|
27 |
# !pip -q install sentencepiece
|
28 |
# load text-to-speech checkpoint and speaker embeddings
|