ataberkkilavuzcu commited on
Commit
64b50bd
·
verified ·
1 Parent(s): e522f34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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
- # Use spk_audio_prompt — this model requires audio prompt
157
  tts_model.infer(
158
  text=payload["text"],
159
- spk_audio_prompt=speaker_file,
160
  output_path=output_file,
161
- use_random=False,
162
- verbose=False,
163
  )
164
 
165
  if not Path(output_file).exists():
166
- raise RuntimeError(f"TTS generation failed — output file not created.")
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(...),