Ifeanyi's picture
Update app.py
e2dfe53 verified
from ImageAnnotator import image_annotate
from PoseAnnotator import pose_annotate
import gradio as gr
objectDetector = gr.Interface(
fn = image_annotate,
theme = "gstaff/sketch",
title="Object Detection",
inputs = [gr.Image(type="filepath",label="Image"), gr.Radio(label="Select Annotator",
choices=["Box","Roundbox","Boxcorner","Color","Circle","Dot","Triangle",
"Ellipse","Percentage","Heatmap","Label","Blur",
"Pixelate","Backgroundcolor"],
value = "Box")],
outputs = gr.Image(label = "Annotated Image"),
examples = [["cars.jpg","Box"],
["colorful-backgrounds-for-laptops.jpg","Roundbox"],
["final_animals-homeschooling_credit-alamy.jpg","Circle"],
["Furry.png","Pixelate"],
["Desktop-Wallpaper-HD4.jpeg","Label"],
["bowling_ball.jpg","Percentage"]]
)
poseDetector = gr.Interface(
fn = pose_annotate,
theme = "gstaff/sketch",
title="Pose Detection",
inputs = [gr.Image(type="filepath",label="Image"), gr.Radio(label="Select Annotator",
choices=["Vertex","Edge","Vertexlabel"],
value = "Vertex")],
outputs = gr.Image(label = "Annotated Image"),
examples = [["Pose.jpg", "Edge"],["black runner.jpg", "Vertexlabel"]]
)
app = gr.TabbedInterface([objectDetector,poseDetector],["Object Detection","Pose Detection"], theme = "gstaff/sketch")
if __name__ == "__main__":
app.launch(mcp_server = True)