Update app.py

app.py

@@ -1,26 +1,17 @@
 import gradio as gr
-import librosa
-import sounddevice as sd
-import numpy as np
 from ttsmms import download, TTS
 
 # Download and load the Swahili TTS model
 dir_path = download("swh", "./data")  # Change "swh" to another language if needed
 tts = TTS(dir_path)
 
-# Function to generate speech from text
+# Function to generate speech from text and return a .wav file
 def text_to_speech(text):
-    result = tts.synthesis(text)
-    audio = result["x"]
-    sample_rate = result["sampling_rate"]
+    wav_path = "./output.wav"
+    tts.synthesis(text, wav_path=wav_path)
+    return wav_path
 
-    # Play generated speech in real-time
-    sd.play(audio, samplerate=sample_rate)
-    sd.wait()
-
-    return audio, sample_rate
-
-# Gradio UI for TTS
+# Gradio UI
 gr.Interface(
     fn=text_to_speech,
     inputs=gr.Text(label="Enter Text"),