from transformers import Pipeline
from chatterbox.turbo import synthesize_turbo

class ChatterboxTurboPipeline(Pipeline):
    def _sanitize_parameters(self, **kwargs):
        return {}, {}, kwargs

    def __call__(self, text, voice_embedding=None, **kwargs):
        audio = synthesize_turbo(
            text=text,
            voice_embedding=voice_embedding,
            **kwargs
        )
        return audio