import torch


def load(pretrained_model_path):
    """
    returns model, tokenizer, image_processor
    """
    pass


def generate(model, tokenizer, prompt, images):
    pass


if __name__ == "__main__":
    model = load("")
    output = generate(
        model,
    )
