import gradio as gr | |
from diffusers import DiffusionPipeline | |
pipeline = DiffusionPipeline.from_pretrained("jax-diffusers-event/canny-coyo1m") | |
def infer(prompt): | |
image = pipeline("prompt").images[0] | |
return image | |
gr.Interface(pipeline, inputs="text", outputs="image").launch() |