File size: 1,009 Bytes
b03e0d7 58ae671 b03e0d7 58ae671 b03e0d7 06521b6 b03e0d7 f8f25ad b03e0d7 f8f25ad b03e0d7 f8f25ad | 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 | import gradio as gr
from image_resizer import ImageResizer
MODEL_PATH = "face_detection_yunet_2023mar.onnx"
image_resizer = ImageResizer(modelPath=MODEL_PATH)
def face_detector(input_image, target_size=512):
target_size = int(target_size)
return image_resizer.resize(input_image, target_size)
inputs = [
gr.Image(sources=["upload", "clipboard"], type="numpy"),
gr.Dropdown(
choices=[512, 768, 1024],
value=512,
allow_custom_value=True,
info="Target size of images",
),
]
outputs = [
gr.Image(label="face detection", format="JPEG"),
gr.Image(label="focused resized", format="JPEG"),
]
demo = gr.Interface(
fn=face_detector,
inputs=inputs,
outputs=outputs,
title="Image Resizer",
api_name="resize",
submit_btn=gr.Button("Resize", variant="primary"),
flagging_mode="never",
)
demo.queue(
max_size=10,
)
app_theme = gr.themes.Monochrome(primary_hue="red")
if __name__ == "__main__":
demo.launch(theme=app_theme)
|