from transformers.tools import load_tool


class EndpointHandler:
    def __init__(self, path=''):
        self.tool = load_tool('huggingface-tools/text-to-video')
        self.tool.setup()

    def __call__(self, data):
        inputs = data.pop("inputs", data)
        return self.tool(**inputs)