Prince9191's picture
Update app.py
8c95b6f verified
raw
history blame contribute delete
642 Bytes
import gradio as gr
from PIL import Image
from object_detection import run_od_pipeline
def detect_objects(image: Image.Image) -> Image.Image:
return run_od_pipeline(image)
# Gradio Interface
title = "🧠 Object Detection using Hugging Face"
description = "Upload an image and run object detection powered by Hugging Face models."
demo = gr.Interface(
fn=detect_objects,
inputs=gr.Image(type="pil", label="Upload an image"),
outputs=gr.Image(type="pil", label="Detected Objects"),
title=title,
description=description,
theme="default",
allow_flagging="never"
)
if __name__ == "__main__":
demo.launch()