Update app.py

app.py

@@ -1,25 +1,22 @@
 import gradio as gr
-from TTS.utils.generic_utils import download_model
-from TTS.utils.io import load_config
-from TTS import TTS
-import numpy as np
-import soundfile as sf
+import torch
+from TTS.api import TTS  # Correct import path for TTS API
 
 # Define paths to the model and configuration
 model_path = "best_model.pth"  # Directory where your model is saved
 config_path = "config.json"  # Configuration file
 
-# Load the model
-config = load_config(config_path)
-tts = TTS(config, model_path=model_path)
+# Initialize TTS with your model and configuration
+tts = TTS(model_path=model_path, config_path=config_path)
 
 def generate_speech(text):
     # Generate speech using the model
-    wav = tts.synthesize(text)
+    wav = tts.tts(text)
     
     # Save the generated audio to a temporary file
     audio_path = "output.wav"
-    sf.write(audio_path, wav, tts.sampling_rate)
+    with open(audio_path, "wb") as f:
+        f.write(wav)
     
     # Read the audio file to return as binary data
     with open(audio_path, "rb") as f: