Update app.py

app.py

@@ -2,15 +2,15 @@ import gradio as gr
 import torchaudio
 import torch
 from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
-from transformers import MMSProcessor, MMSForConditionalGeneration
+from transformers import AutoProcessor, AutoModelForSeq2SeqLM
 from transformers import Wav2Vec2Processor, Wav2Vec2ForSequenceClassification
 
 # Load the models
 asr_model = Wav2Vec2ForCTC.from_pretrained("facebook/mms-1b-all")
 asr_processor = Wav2Vec2Processor.from_pretrained("facebook/mms-1b-all")
 
-tts_model = MMSForConditionalGeneration.from_pretrained("facebook/mms-tts")
-tts_processor = MMSProcessor.from_pretrained("facebook/mms-tts")
+tts_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mms-tts")
+tts_processor = AutoProcessor.from_pretrained("facebook/mms-tts")
 
 lid_model = Wav2Vec2ForSequenceClassification.from_pretrained("facebook/mms-lid-1024")
 lid_processor = Wav2Vec2Processor.from_pretrained("facebook/mms-lid-1024")