Update app.py

app.py

@@ -19,8 +19,8 @@ translate_token_id = all_special_ids[-6]
 
 # --- FastSpeech2 (TTS) Setup ---
 TTS_MODEL_NAME = "facebook/fastspeech2-en-ljspeech"
-tts_processor = AutoProcessor.from_pretrained(TTS_MODEL_NAME)
-tts_model = SpeechEncoderDecoderModel.from_pretrained(TTS_MODEL_NAME)
+tts_processor = AutoProcessor.from_pretrained(TTS_MODEL_NAME, trust_remote_code=True)
+tts_model = SpeechEncoderDecoderModel.from_pretrained(TTS_MODEL_NAME, trust_remote_code=True)
 tts_device = "cuda" if torch.cuda.is_available() else "cpu"  # Use CUDA if available
 tts_model = tts_model.to(tts_device)