Update detection.py

detection.py

@@ -3,7 +3,7 @@ import IPython
 from PIL import ImageColor
 from ultralytics import YOLO
 
-class ObjectDetection:
+class ObjectDetectionction:
     def __init__(self, model_name='Yolov8'):
         self.model_name = model_name
         self.model = self.load_model()
@@ -17,9 +17,9 @@ class ObjectDetection:
     def v8_score_frame(self, frame):
         results = self.model(frame)
 
-        labels = results.names[results.pred[..., -1].argmax(-1)]  # Get class labels
-        confidences = results.pred[..., -2].max(-1)  # Get confidences
-        coords = results.pred[..., :-2]  # Get coordinates
+        labels = results.names  # Get class labels
+        confidences = results.probs  # Get confidences
+        coords = results.boxes  # Get coordinates
 
         return labels, confidences, coords