File size: 320 Bytes
5e081d2 66b4522 5e081d2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | import gradio as gr
from diffusers import StableDiffusionPipeline
import torch
pipe = StableDiffusionPipeline.from_pretrained(
"runwayml/stable-diffusion-v1-5",
).to("cpu")
def generate(prompt):
image = pipe(prompt).images[0]
return image
gr.Interface(fn=generate, inputs="text", outputs="image").launch() |