Update app.py

app.py

@@ -5,6 +5,10 @@ from datasets import load_dataset
 
 from transformers import SpeechT5ForTextToSpeech, SpeechT5HifiGan, SpeechT5Processor, pipeline
 
+## Imports for MMS
+# from transformers import VitsModel, VitsTokenizer
+
+
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
@@ -19,12 +23,13 @@ asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base",
 
 
 # For Dutch
-model_id = 'sanchit-gandhi/speecht5_tts_vox_nl'
 
+##### speecht5 #####
+model_id = 'sanchit-gandhi/speecht5_tts_vox_nl'
 processor = SpeechT5Processor.from_pretrained(model_id)
 model = SpeechT5ForTextToSpeech.from_pretrained(model_id)
 
-vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan")
+vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan").to(device)