ylacombe HF staff commited on
Commit
a3c43e8
1 Parent(s): 15000e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ num_speakers = pipe.model.config.num_speakers
28
  # Inference
29
  def generate_audio(text, spkr_id):
30
 
31
- forward_params = {"spkr_id": spkr_id}
32
  output = pipe(text, forward_params=forward_params)
33
 
34
  return (output["sampling_rate"], output["audio"].squeeze())
 
28
  # Inference
29
  def generate_audio(text, spkr_id):
30
 
31
+ forward_params = {"speaker_id": spkr_id}
32
  output = pipe(text, forward_params=forward_params)
33
 
34
  return (output["sampling_rate"], output["audio"].squeeze())