Thomcles commited on
Commit
87cff19
·
verified ·
1 Parent(s): e8d6884

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -61,7 +61,11 @@ def load_tts_model(repo: str, checkpoint_file: str, device: str) -> ChatterboxTT
61
 
62
  def synthesize_speech(model: ChatterboxTTS, text: str, audio_prompt_path:str, **kwargs) -> torch.Tensor:
63
  with torch.inference_mode():
64
- return model.generate(text, audio_prompt_path, **kwargs)
 
 
 
 
65
 
66
  def save_audio(waveform: torch.Tensor, path: str, sample_rate: int):
67
  sf.write(path, waveform.squeeze().cpu().numpy(), sample_rate)
 
61
 
62
  def synthesize_speech(model: ChatterboxTTS, text: str, audio_prompt_path:str, **kwargs) -> torch.Tensor:
63
  with torch.inference_mode():
64
+ return model.generate(
65
+ text=text,
66
+ audio_prompt_path=audio_prompt_path,
67
+ **kwargs
68
+ )
69
 
70
  def save_audio(waveform: torch.Tensor, path: str, sample_rate: int):
71
  sf.write(path, waveform.squeeze().cpu().numpy(), sample_rate)