Create app.py

app.py

@@ -0,0 +1,76 @@
+import supervision as sv
+import gradio as gr
+from ultralytics import YOLO
+import sahi
+
+
+
+# Images
+sahi.utils.file.download_from_url(
+    "https://transform.roboflow.com/zD7y6XOoQnh7WC160Ae7/4d51f997137c0dca78fa2c9154e0b51a/thumb.jpg",
+    "f1.jpg",
+)
+sahi.utils.file.download_from_url(
+    "https://transform.roboflow.com/zD7y6XOoQnh7WC160Ae7/48174c7c26c2cbca52b084ebbb03d215/thumb.jpg",
+    "f2.jpg",
+)
+sahi.utils.file.download_from_url(
+    "https://transform.roboflow.com/zD7y6XOoQnh7WC160Ae7/3d1f22e387164a6719995aa0d9dc16a1/thumb.jpg",
+    "f3.jpg",
+)
+
+
+
+
+annotatorbbox = sv.BoxAnnotator()
+annotatormask=sv.MaskAnnotator()
+
+
+def yolov8_inference(
+    image: gr.inputs.Image = None,
+    model_name: gr.inputs.Dropdown = None,
+    image_size: gr.inputs.Slider = 320,
+    conf_threshold: gr.inputs.Slider = 0.25,
+    iou_threshold: gr.inputs.Slider = 0.45,
+):
+
+
+    model = YOLO("/content/segment/train/weights/best.pt")
+
+    results = model(image,conf=conf_threshold,iou=iou_threshold ,imgsz=320)[0]
+    detections = sv.Detections.from_yolov8(results)
+    annotated_image = annotatorbbox.annotate(scene=image, detections=detections)
+    annotated_image = annotatormask.annotate(scene=annotated_image, detections=detections)
+
+
+
+
+    return annotated_image
+
+image_input = gr.inputs.Image()  # Adjust the shape according to your requirements
+
+inputs = [
+    gr.inputs.Image(label="Input Image"),
+    gr.Slider(
+        minimum=0.0, maximum=1.0, value=0.25, step=0.05, label="Confidence Threshold"
+    ),
+    gr.Slider(minimum=0.0, maximum=1.0, value=0.45, step=0.05, label="IOU Threshold"),
+]
+
+outputs = gr.Image(type="filepath", label="Output Image")
+title = "Ultralytics YOLOv8 Segmentation Demo"
+import os
+examples = [
+    ["f1.jpg", 0.6, 0.45],
+    ["f2.jpg", 0.25, 0.45],
+    ["f3.jpg", 0.25, 0.45],
+]
+demo_app = gr.Interface(examples=examples,
+    fn=yolov8_inference,
+    inputs=inputs,
+    outputs=outputs,
+    title=title,
+    cache_examples=True,
+    theme="default",
+)
+demo_app.launch(debug=True, enable_queue=True)