File size: 494 Bytes
431cb87 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import gradio as gr
import numpy as np
def percent_of_pixels_selected(image):
mask = image["layers"][0]
mask_bool = mask[:,:,3] > 0
return f"{round(np.sum(mask_bool) / mask_bool.size * 100, 2)}%"
image_editor = gr.Sketchpad(
type="numpy",
)
output_image = gr.Label(
label="Percent of Pixels Selected"
)
demo = gr.Interface(
fn=percent_of_pixels_selected,
inputs=image_editor,
outputs=output_image,
live=True
)
if __name__ == "__main__":
demo.launch() |