Adding channel for GradCAM

app.py

@@ -75,6 +75,7 @@ def inference(input_img, conf_thres, iou_thres, is_false_detection_images=True,
         # Convert to tensor                          
         ToTensorV2()])
     im_resized = transforms(image = im_resized)['image']
+    im_resized = im_resized.unsqueeze(0)
     stride, names, pt = model.stride, model.names, model.pt
     imgsz = check_img_size((640, 640), s=stride)  # check image size