Fix: version dependency

app.py

@@ -33,14 +33,12 @@ def parse_detections(results, yolo_version):
     elif yolo_version == "yolov8":
         # For YOLOv8
         for result in results:
-            for box in result.boxes:
-                xmin, ymin, xmax, ymax = box.xyxy.tolist()
-                confidence = box.conf.item()
-                class_id = int(box.cls.item())
+            for box in result.boxes.data.tolist():  # Access box data directly
+                xmin, ymin, xmax, ymax, confidence, class_id = box
                 if confidence > 0.2:
                     boxes.append((xmin, ymin, xmax, ymax))
-                    colors.append(COLORS[class_id % len(COLORS)])
-                    names.append(result.names[class_id])
+                    colors.append(COLORS[int(class_id) % len(COLORS)])
+                    names.append(result.names[int(class_id)])
 
     return boxes, colors, names
 
@@ -89,7 +87,7 @@ def process_image(image, yolo_versions=["yolov5"]):
         # Parse detections using updated function
         boxes, colors, names = parse_detections(results[0], yolo_version)
         
-        detections_img = draw_detections(boxes, colors.copy(), names.copy(), rgb_img.copy())
+        detections_img = draw_detections(boxes.copy(), colors.copy(), names.copy(), rgb_img.copy())
 
         # Grad-CAM visualization
         target_layers = [model.model.model[-1]]  # Use last layer as target layer for Grad-CAM