Update app.py

app.py

@@ -14,6 +14,7 @@ from dotenv import load_dotenv
 import tempfile
 import spaces
 from TTS.api import TTS
+from TTS.utils.synthesizer import Synthesizer
 
 # Load environment variables
 load_dotenv()
@@ -58,13 +59,18 @@ def generate_script(user_prompt: str, model_id: str, token: str, duration: int):
 @spaces.GPU(duration=300)
 def generate_voice(script: str, speaker: str = "default"):
     try:
-        # Load the TTS model
-        tts = TTS(model_name="tts_models/en/ljspeech/tacotron2-DDC", gpu=torch.cuda.is_available())
-        
-        # Generate the speech audio file
+        # Load TTS model
+        tts_model_path = "tts_models/en/ljspeech/tacotron2-DDC"
+        vocoder_model_path = "vocoder_models/en/ljspeech/hifigan_v2"
+        synthesizer = Synthesizer(tts_model_path, vocoder_model_path)
+
+        # Generate audio
+        wav = synthesizer.tts(script)
+
+        # Save output to a file
         output_path = f"{tempfile.gettempdir()}/generated_voice.wav"
-        tts.tts_to_file(text=script, file_path=output_path, speaker=speaker)
-        
+        synthesizer.save_wav(wav, output_path)
+
         return output_path
     except Exception as e:
         return f"Error generating voice-over: {e}"