Update app.py

app.py

@@ -102,6 +102,7 @@ class TTSPayload(BaseModel):
 # TTS ENDPOINT
 # ===============================
 @app.post("/tts")
+
 def generate_tts(payload: TTSPayload):
     model = get_or_load_model()
 
@@ -117,10 +118,11 @@ def generate_tts(payload: TTSPayload):
 
     # CPU-safe inference
     with torch.no_grad():
-        sr, wav = model.generate(
+        wav = model.generate(
             final_text[:300],
             language_id=payload.language_id,
         )
+        sr = model.sr  # <-- sample rate from model property
 
     # Convert numpy -> WAV bytes
     buf = io.BytesIO()
@@ -134,6 +136,7 @@ def generate_tts(payload: TTSPayload):
         "audio_base64": base64.b64encode(audio_bytes).decode("utf-8")
     }
 
+
 # ===============================
 # RUN: uvicorn app:app --host 0.0.0.0 --port 7860
 # ===============================