jbrinkma's picture
setup app layout
a747dc5
raw history blame
No virus
594 Bytes
import gradio as gr
def segment_image(input_image):
output_image = input_image
return output_image
with gr.Blocks() as demo:
with gr.Row():
input_image = gr.Image(label='Input Image')
output_image = gr.Image(label='Output Image')
button = gr.Button('Mask Image')
button.click(segment_image, inputs=[input_image], outputs=output_image)
gr.Examples(
examples = [
['./imgs/cat.jpg']
],
inputs=[input_image],
outputs=[output_image],
fn=segment_image,
cache_examples=True
)
demo.launch()