import torchaudio as ta
from chatterbox.mtl_tts import ChatterboxMultilingualTTS

# English example
model = ChatterboxMultilingualTTS.from_pretrained(device="cuda")
text = "Wat een sukkel!"
wav = model.generate(text, language_id="nl", audio_prompt_path="./ej-audio-input-sample.wav", exaggeration=1.2)
filename = "ej-output.wav"
ta.save(filename, wav, model.sr)