Minor Changes

app.py

@@ -6,10 +6,10 @@ import supervision as sv
 from inference import get_model
 
 MARKDOWN = """
-<h1 style='text-align: center'>Detect Something 💫</h1>
-Welcome to Detect Something! Just a simple demo to showcase the detection capabilities of various YOLOv8 models. 🚀🔍👀
+<h1 style='text-align: center'>Segment Something 🖼️</h1>
+Welcome to Segment Something! Just a simple demo to showcase the instance segmentation capabilities of various YOLOv8 segmentation models. 🚀🔍👀
 
-A simple project just for fun for on the go object detection. 🎉
+A simple project just for fun for on the go instance segmentation. 🎉
 
 Inspired from YOLO-ARENA by SkalskiP. 🙏
 
@@ -23,12 +23,12 @@ IMAGE_EXAMPLES = [
     ['https://media.roboflow.com/supervision/image-examples/basketball-1.png', 0.3, 0.3, 0.3],
 ]
 
-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")
+YOLO_V8N_MODEL = get_model(model_id="yolov8n-seg.pt")
+YOLO_V8S_MODEL = get_model(model_id="yolov8s-seg.pt")
+YOLO_V8M_MODEL = get_model(model_id="yolov8m-seg.pt")
 
 LABEL_ANNOTATORS = sv.LabelAnnotator(text_color=sv.Color.black())
-BOUNDING_BOX_ANNOTATORS = sv.BoundingBoxAnnotator()
+MASK_ANNOTATORS = sv.MaskAnnotator()
 
 
 def detect_and_annotate(
@@ -59,7 +59,7 @@ def detect_and_annotate(
     ]
 
     annotated_image = input_image.copy()
-    annotated_image = BOUNDING_BOX_ANNOTATORS.annotate(
+    annotated_image = MASK_ANNOTATORS.annotate(
         scene=annotated_image, detections=detections)
     annotated_image = LABEL_ANNOTATORS.annotate(
         scene=annotated_image, detections=detections, labels=labels)