Fix: resolved model access methodology

app.py

@@ -74,7 +74,12 @@ 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[-1]]  # Fix: Access the correct last layer in YOLOv8
+        
+        # YOLOv8: Extract last layer by model.model[-1] would not work; use the following:
+        if isinstance(model.model, torch.nn.Sequential):
+            target_layers = [model.model[-1]]  # This assumes model layers are in a Sequential container
+        else:
+            target_layers = [model.model[-2]]  # Use an appropriate layer
 
         # Run YOLO detection
         results = model([rgb_img])