Update app.py

app.py

@@ -1,22 +1,23 @@
-import streamlit as st
-from PIL import Image
-from object_detection import run_od_pipeline
-
-# Streamlit Page Setup
-st.set_page_config(page_title="🧠 Object Detection App", layout="centered")
-st.title("🧠 Object Detection using Hugging Face")
-
-st.markdown("Upload an image to run object detection:")
-
-# Upload image
-uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
-
-if uploaded_file:
-    image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
-
-    # Run object detection
-    st.write("Running object detection...")
-    detected_img = run_od_pipeline(image)
-
-    st.image(detected_img, caption="Detected Objects", use_column_width=True)
+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()