Create app.py

app.py

@@ -0,0 +1,35 @@
+# app.py
+
+from transformers import pipeline
+import gradio as gr
+from PIL import Image, ImageDraw
+
+# Load the object detection pipeline with YOLO
+detector = pipeline("object-detection", model="hustvl/yolos-tiny")
+
+# Define the detection function
+def detect_objects(img):
+    results = detector(img)
+
+    # Draw bounding boxes
+    draw = ImageDraw.Draw(img)
+    for obj in results:
+        box = obj["box"]
+        label = f"{obj['label']} ({obj['score']:.2f})"
+        draw.rectangle([box["xmin"], box["ymin"], box["xmax"], box["ymax"]], outline="red", width=3)
+        draw.text((box["xmin"], box["ymin"] - 10), label, fill="red")
+    
+    return img
+
+# Create the Gradio interface
+interface = gr.Interface(
+    fn=detect_objects,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
+    title="YOLO Object Detection",
+    description="Upload an image to detect objects using a YOLOS-Tiny model."
+)
+
+# Launch the app
+if __name__ == "__main__":
+    interface.launch()