changed3 app.py

app.py

@@ -1,36 +1,38 @@
 import json
-import os
+from pathlib import Path
 
 import numpy as np
 import gradio as gr
-from TTS.utils.synthesizer import Synthesizer
+from TTS.api import TTS
 
 # ---------- Paths ----------
-MODEL_PATH = "checkpoint_80000.pth"   # or "best_model_23206.pth"
+MODEL_PATH = "checkpoint_80000.pth"     # or checkpoint_70000.pth if that's the good one
 CONFIG_PATH = "config.json"
 
-# ---------- Load config to get sample rate ----------
-with open(CONFIG_PATH, "r", encoding="utf-8") as f:
-    cfg = json.load(f)
+run_dir = Path(".")
 
-SAMPLE_RATE = cfg.get("audio", {}).get("sample_rate", 24000)
-
-# ---------- Load Coqui TTS Synthesizer ----------
-synthesizer = Synthesizer(
-    tts_checkpoint=MODEL_PATH,
-    tts_config_path=CONFIG_PATH,
-    use_cuda=False,   # Set True only if you enable GPU in the Space
+# ---------- Load model ----------
+# In Spaces we will force CPU (gpu=False)
+tts = TTS(
+    model_path=str(run_dir / MODEL_PATH),
+    config_path=str(run_dir / CONFIG_PATH),
+    progress_bar=False,
+    gpu=False,
 )
 
+# Sample rate from the loaded synthesizer
+SAMPLE_RATE = tts.synthesizer.output_sample_rate
+
 # ---------- Inference function ----------
 def tts_generate(text: str):
-    if not text.strip():
+    text = text.strip()
+    if not text:
         return None
 
-    # Whatever Coqui returns (list, numpy array, etc.)
-    wav = synthesizer.tts(text)
+    # This is the same as your local script
+    wav = tts.tts(text)
 
-    # Convert to a 1D numpy array for Gradio
+    # Ensure numpy array (1D) for Gradio
     wav = np.asarray(wav, dtype="float32").flatten()
 
     # Gradio Audio(type="numpy") expects (sample_rate, np.ndarray)
@@ -41,15 +43,15 @@ demo = gr.Interface(
     fn=tts_generate,
     inputs=gr.Textbox(
         label="Input text",
-        placeholder="Type Sinhala text here…(use Sinhala script, max 120 characters)",
+        placeholder="Type Sinhala text here…",
         lines=3,
     ),
     outputs=gr.Audio(
         label="Generated speech",
         type="numpy",
     ),
-    title="Sinhala TTS",
-    description="Sinhala TTS model- Research-stage model",
+    title="Sinhala TTS ",
+    description="Sinhala TTS model research-stage model.",
 )
 
 if __name__ == "__main__":