Fix:update access lyrs.

app.py

@@ -75,8 +75,10 @@ def process_image(image, yolo_versions=["yolov5"]):
     for yolo_version in yolo_versions:
         # Load the model based on YOLO version
         model = load_yolo_model(yolo_version)
-        target_layers = [model.model.model.model[-2]]  # Assumes last layer is used for Grad-CAM
-
+        if yolo_version == "yolov5":
+            target_layers = [model.model.model.model[-2]]  # Assumes last layer is used for Grad-CAM
+        else:
+            target_layers = [model.model[-2]]
         # Run YOLO detection
         results = model([rgb_img])
         boxes, colors, names = parse_detections(results)