unijoh commited on
Commit
7bce9ab
1 Parent(s): 0fa99f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -9,8 +9,9 @@ from transformers import Wav2Vec2Processor, Wav2Vec2ForSequenceClassification
9
  asr_model = Wav2Vec2ForCTC.from_pretrained("facebook/mms-1b-all")
10
  asr_processor = Wav2Vec2Processor.from_pretrained("facebook/mms-1b-all")
11
 
12
- tts_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mms-tts")
13
- tts_processor = AutoProcessor.from_pretrained("facebook/mms-tts")
 
14
 
15
  lid_model = Wav2Vec2ForSequenceClassification.from_pretrained("facebook/mms-lid-1024")
16
  lid_processor = Wav2Vec2Processor.from_pretrained("facebook/mms-lid-1024")
 
9
  asr_model = Wav2Vec2ForCTC.from_pretrained("facebook/mms-1b-all")
10
  asr_processor = Wav2Vec2Processor.from_pretrained("facebook/mms-1b-all")
11
 
12
+ # Correct TTS model path
13
+ tts_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mms-tts/models/fao")
14
+ tts_processor = AutoProcessor.from_pretrained("facebook/mms-tts/models/fao")
15
 
16
  lid_model = Wav2Vec2ForSequenceClassification.from_pretrained("facebook/mms-lid-1024")
17
  lid_processor = Wav2Vec2Processor.from_pretrained("facebook/mms-lid-1024")