crcdng commited on
Commit
d88edc1
·
1 Parent(s): 4a07a83

added .to(device)

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base",
17
  processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
18
 
19
  # model = SpeechT5ForTextToSpeech.from_pretrained("microsoft/speecht5_tts").to(device)
20
- model = VitsModel.from_pretrained("Matthijs/mms-tts-nld")
21
- tokenizer = VitsTokenizer.from_pretrained("Matthijs/mms-tts-nld")
22
 
23
  vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
24
 
 
17
  processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
18
 
19
  # model = SpeechT5ForTextToSpeech.from_pretrained("microsoft/speecht5_tts").to(device)
20
+ model = VitsModel.from_pretrained("Matthijs/mms-tts-nld").to(device)
21
+ tokenizer = VitsTokenizer.from_pretrained("Matthijs/mms-tts-nld").to(device)
22
 
23
  vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)
24