Update display.py

display.py

@@ -14,6 +14,7 @@ def inference(image: np.ndarray, iou_thresh: float = 0.75, thresh: float = 0.75,
     # iou_thresh = 0.75
     # thresh = 0.75
     scaled_anchors  =  config.SCALED_ANCHORS
+    cam = YoloCAM(model=model, target_layers=[model.layers[-2]], use_cuda=False)
     transforms =  A.Compose(
     [
         A.LongestMaxSize(max_size=config.IMAGE_SIZE),