import videogen_hub


class VideogenHubModel():
    def __init__(self, model_name):
        self.model = videogen_hub.load(model_name)

    def __call__(self, *args, **kwargs):
        return self.model.infer_one_video(*args, **kwargs)

def load_videogenhub_model(model_name):
    return VideogenHubModel(model_name)