Update app.py

app.py

@@ -7,20 +7,14 @@ from ultralytics import YOLO
 # Load the YOLOv8 model
 model = YOLO('best.pt')
 
-# Define a list of colors for different classes
+# Define a list of colors for the 6 classes
 colors = [
     (255, 0, 0),     # Red
     (0, 255, 0),     # Green
     (0, 0, 255),     # Blue
     (255, 255, 0),   # Cyan
     (255, 0, 255),   # Magenta
-    (0, 255, 255),   # Yellow
-    (128, 0, 0),     # Maroon
-    (0, 128, 0),     # Olive
-    (128, 128, 0),   # Teal
-    (0, 0, 128),     # Navy
-    (128, 0, 128),   # Purple
-    (0, 128, 128)    # Aqua
+    (0, 255, 255)    # Yellow
 ]
 
 def detect_objects(image):