Create app.py

app.py

@@ -0,0 +1,73 @@
+import gradio as gr
+from PIL import Image, ImageDraw
+
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+
+
+# model_path = ("../models/models--facebook--detr-resnet-50/snapshots/1d5f47bd3bdd2c4bbfa585418ffe6da5028b4c0b")
+
+# object_detector = pipeline("object-detection", model=model_path)
+
+object_detector = pipeline("object-detection", model="facebook/detr-resnet-50")
+
+
+
+# raw_image = Image .open("../Files/child dog ball.jpg")
+# output = object_detector(raw_image)
+# print(output)
+
+def draw_bounding_boxes(image, detections):
+    """
+    Draws bounding boxes around detected objects on the image.
+
+    Parameters:
+    image (PIL.Image): The input image.
+    detections (list): A list of dictionaries containing detection results.
+
+    Returns:
+    PIL.Image: The image with bounding boxes drawn around detected objects.
+    """
+    # Create an ImageDraw object to draw on the image
+    draw = ImageDraw.Draw(image)
+
+    # Iterate over each detected object
+    for detection in detections:
+        box = detection['box']
+        label = detection['label']
+        score = detection['score']
+
+        # Define the coordinates of the bounding box
+        xmin, ymin, xmax, ymax = box['xmin'], box['ymin'], box['xmax'], box['ymax']
+
+        # Draw the bounding box on the image
+        draw.rectangle([xmin, ymin, xmax, ymax], outline="red", width=2)
+
+        # Draw the label and score on the image
+        text = f"{label} ({score:.2f})"
+        draw.text((xmin, ymin - 10), text, fill="red")
+
+    return image
+#raw_image = Image .open("image")
+def detect_object(image):
+    raw_image = image
+    output = object_detector(raw_image)
+    processed_image = draw_bounding_boxes(raw_image, output)
+    return processed_image
+
+# print(output)
+
+
+gr.close_all()
+
+# Create the Gradio interface
+demo = gr.Interface(
+    fn=detect_object,
+    inputs=gr.Image(label="Select an Image", type="pil"),
+    outputs=gr.Image(label="Processed Image", type="pil"),
+    title="GenAI Project 6: Object Detector",
+    description="THIS APPLICATION WILL BE USED TO DETECT OBJECTS IN THE PROVIDED INPUT"
+)
+
+# Launch the Gradio interface
+demo.launch()