Spaces:
Sleeping
Sleeping
File size: 1,786 Bytes
1aa3766 |
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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import pathlib
import cv2
import gradio as gr
import numpy as np
MAX_W, MAX_H = 600, 400
def update_input_image(image: np.ndarray):
if image is None:
return gr.Image.update(value=None)
h, w = image.shape[:2]
scale = min(MAX_W / w, MAX_H / h)
if scale < 1:
image = cv2.resize(image, None, fx=scale, fy=scale)
return gr.Image.update(value=image)
def set_example_image(example):
return [
gr.Image.update(value=example[0]),
gr.Image.update(value=example[0])
]
with gr.Blocks() as demo:
with gr.Row():
with gr.Column():
input_image1 = gr.Image(label='Input',
type='numpy',
shape=(MAX_W, MAX_H))
with gr.Column():
output_image1 = gr.Image(label='Output')
with gr.Row():
with gr.Column():
input_image2 = gr.Image(label='Input', type='numpy')
with gr.Column():
output_image2 = gr.Image(label='Output')
button = gr.Button('Run')
with gr.Row():
paths = sorted(pathlib.Path('images').rglob('*.jpg'))
examples = gr.Dataset(components=[input_image1, input_image2],
samples=[[path.as_posix(),
path.as_posix()] for path in paths])
input_image2.change(fn=update_input_image,
inputs=input_image2,
outputs=input_image2)
button.click(fn=lambda x: x, inputs=input_image1, outputs=output_image1)
button.click(fn=lambda x: x, inputs=input_image2, outputs=output_image2)
examples.click(fn=set_example_image,
inputs=[examples],
outputs=[input_image1, input_image2])
demo.launch()
|