from pipeline import Pipeline
from device import device, torch_dtype
from diffusers.utils import load_image

def main():
    p = Pipeline(device, torch_dtype)
    params = Pipeline.InputParams()
    image = load_image("https://hf.co/datasets/huggingface/documentation-images/resolve/main/diffusers/input_image_vermeer.png")
    p.predict(params, image).show()

if __name__ == "__main__":
    main()