Update app.py

app.py

@@ -1,13 +1,21 @@
-from fastapi import FastAPI, Body
+from fastapi import FastAPI
+from pydantic import BaseModel
 from huggingface_hub import snapshot_download
 from transformers import pipeline
 import os
 
+class TTSRequest(BaseModel):
+    text: str
+
 app = FastAPI()
-model_dir = snapshot_download("blacknight3113/edith", use_auth_token=True, repo_type="model")
+model_dir = snapshot_download(
+    "blacknight3113/edith",
+    repo_type="model",
+    use_auth_token=True
+)
 tts = pipeline("text-to-speech", model=model_dir)
 
 @app.post("/tts")
-async def tts_api(text: str = Body(..., embed=True)):
-    out = tts(text, forward_params={"do_sample": True})
+async def tts_api(req: TTSRequest):
+    out = tts(req.text, forward_params={"do_sample": True})
     return {"sr": out["sampling_rate"], "audio": out["audio"].tolist()}