Attempting Detection

app.py

@@ -6,7 +6,7 @@ import supervision as sv
 from inference import get_model
 
 MARKDOWN = """
-<h1 style='text-align: center'>Segment Something 💫</h1>
+<h1 style='text-align: center'>Detect Something 💫</h1>
 Welcome to Segment Something! Your on the go demo for instance segmentation. 🚀
 
 <h2 style='text-align: center'>Matthias Bartolo</h2>
@@ -21,12 +21,12 @@ IMAGE_EXAMPLES = [
     ['https://media.roboflow.com/supervision/image-examples/basketball-1.png', 0.3, 0.3, 0.1],
 ]
 
-YOLO_V8N_MODEL = get_model(model_id="yolov8n-seg-640")
-YOLO_V8S_MODEL = get_model(model_id="yolov8s-seg-640")
-YOLO_V8M_MODEL = get_model(model_id="yolov8m-seg-640")
+YOLO_V8N_MODEL = get_model(model_id="yolov8n-640")
+YOLO_V8S_MODEL = get_model(model_id="yolov8s-640")
+YOLO_V8M_MODEL = get_model(model_id="yolov8m-640")
 
 LABEL_ANNOTATORS = sv.LabelAnnotator(text_color=sv.Color.black())
-INSTANCE_SEGMENTATION_ANNOTATORS = sv.InstanceSegmentationAnnotator()
+BOUNDING_BOX_ANNOTATORS = sv.BoundingBoxAnnotator()
 
 
 def detect_and_annotate(
@@ -57,7 +57,7 @@ def detect_and_annotate(
     ]
 
     annotated_image = input_image.copy()
-    annotated_image = INSTANCE_SEGMENTATION_ANNOTATORS.annotate(
+    annotated_image = BOUNDING_BOX_ANNOTATORS.annotate(
         scene=annotated_image, detections=detections)
     annotated_image = LABEL_ANNOTATORS.annotate(
         scene=annotated_image, detections=detections, labels=labels)