Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -153,24 +153,24 @@ def _run_generate_job(job_id: str, payload: Dict[str, str]):
|
|
| 153 |
|
| 154 |
output_file = os.path.join(tempfile.gettempdir(), f"indextts2-{uuid.uuid4()}.wav")
|
| 155 |
|
| 156 |
-
#
|
| 157 |
tts_model.infer(
|
| 158 |
text=payload["text"],
|
| 159 |
-
|
| 160 |
output_path=output_file,
|
| 161 |
-
use_random=False,
|
| 162 |
-
verbose=False,
|
| 163 |
)
|
| 164 |
|
| 165 |
if not Path(output_file).exists():
|
| 166 |
-
raise RuntimeError(
|
| 167 |
|
| 168 |
_set_job(job_id, status="completed", output_file=output_file)
|
|
|
|
| 169 |
except Exception as exc:
|
| 170 |
_cleanup_files(speaker_file or "", output_file or "")
|
| 171 |
_set_job(job_id, status="error", error=str(exc))
|
| 172 |
|
| 173 |
|
|
|
|
| 174 |
@app.post("/generate")
|
| 175 |
def generate(
|
| 176 |
payload: GenerateRequest = Body(...),
|
|
|
|
| 153 |
|
| 154 |
output_file = os.path.join(tempfile.gettempdir(), f"indextts2-{uuid.uuid4()}.wav")
|
| 155 |
|
| 156 |
+
# FIXED ARGUMENT NAME
|
| 157 |
tts_model.infer(
|
| 158 |
text=payload["text"],
|
| 159 |
+
speaker_prompt=speaker_file,
|
| 160 |
output_path=output_file,
|
|
|
|
|
|
|
| 161 |
)
|
| 162 |
|
| 163 |
if not Path(output_file).exists():
|
| 164 |
+
raise RuntimeError("TTS generation failed — output file not created.")
|
| 165 |
|
| 166 |
_set_job(job_id, status="completed", output_file=output_file)
|
| 167 |
+
|
| 168 |
except Exception as exc:
|
| 169 |
_cleanup_files(speaker_file or "", output_file or "")
|
| 170 |
_set_job(job_id, status="error", error=str(exc))
|
| 171 |
|
| 172 |
|
| 173 |
+
|
| 174 |
@app.post("/generate")
|
| 175 |
def generate(
|
| 176 |
payload: GenerateRequest = Body(...),
|