import gradio as gr def generate(image_editor): return image_editor with gr.Blocks() as demo: gr.Markdown(""" # Inpainting Sketch Pad by [Tony Assi](https://www.tonyassi.com/) """) with gr.Row(): with gr.Column(): version_gallery = gr.Gallery() restore_button = gr.Button("Restore Version") with gr.Column(): sketch_pad = gr.ImageEditor() generate_button = gr.Button("Generate") with gr.Row(): out1 = gr.Image() out2 = gr.Image() out3 = gr.Image() generate_button.click(fn=generate, inputs=sketch_pad, outputs=sketch_pad) demo.launch()