Update README.md
Browse files
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(
|
|
|
|
|
|
|
|
|
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)
|