Spaces:
Sleeping
Sleeping
File size: 754 Bytes
4f2e372 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import gradio as gr
from diffusers import DDPMPipeline
model = 'alexktrs/CumulusCloudsGenerator'
device='cpu'
generator = DDPMPipeline.from_pretrained(model)
generator.to(device)
def generate(num_inference_steps):
image1 = generator(num_inference_steps=num_inference_steps).images[0]
image2 = generator(num_inference_steps=num_inference_steps).images[0]
return [image1, image2]
with gr.Blocks() as demo:
gr.Markdown("""
# Generate Cumulus Clouds
""")
gallery=gr.Gallery(type="pil")
slider=gr.Slider(label='Inference Steps', minimum=0, maximum=100)
btn = gr.Button("Generate Clouds")
btn.click(fn=generate, inputs=slider, outputs=gallery)
demo.launch() |