Fix.

app.py

@@ -16,8 +16,12 @@ from ultralytics import YOLO
 COLORS = np.random.uniform(0, 255, size=(80, 3))
 
 # Function to parse YOLO detections
-def parse_detections(results):
-    detections = results.pandas().xyxy[0].to_dict()
+def parse_detections(results, yolo_version):
+    if yolo_version == "yolov5":
+        detections = results.pandas().xyxy[0].to_dict()
+    else:
+        detections = results[0].pandas().xyxy[0].to_dict()
+
     boxes, colors, names = [], [], []
     for i in range(len(detections["xmin"])):
         confidence = detections["confidence"][i]
@@ -81,7 +85,7 @@ def process_image(image, yolo_versions=["yolov5"]):
             target_layers = [model.model.model[-1]]
         # Run YOLO detection
         results = model([rgb_img])
-        boxes, colors, names = parse_detections(results)
+        boxes, colors, names = parse_detections(results, yolo_version)
         detections_img = draw_detections(boxes, colors, names, rgb_img.copy())
 
         # Grad-CAM visualization