CloudGenerator / app.py
alexktrs's picture
Upload 2 files
4f2e372
raw
history blame
754 Bytes
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()