Spaces:
Sleeping
Sleeping
| 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() | |