File size: 399 Bytes
70de1d6
 
6776a75
70de1d6
 
 
6776a75
 
 
70de1d6
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
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()