Update app.py

app.py

@@ -80,7 +80,7 @@ def process_image_and_plot(image,iou_threshold=0.5, threshold=0.4):
         bboxes[0], iou_threshold=iou_threshold, threshold=threshold, box_format="midpoint",
     )
     fig = plot_image(transformed_image[0].permute(1, 2, 0), nms_boxes)
-    cam = YoloCAM(model=model, target_layers=model.model.layers[-2], use_cuda=False)
+    cam = YoloCAM(model=model, target_layers=[model.model.layers[-2]], use_cuda=False)
 
     grayscale_cam = cam(transformed_image, scaled_anchors)[0, :, :]
     img = cv2.resize(image, (416, 416))