debug: stack dimension issue

yolov5.py

@@ -81,6 +81,7 @@ def xai_yolov5(image):
     im = cv2.imread("test_w.png")
     
     # Combine results
+    print('shapes........', image.shape, detections_img.shape, renormalized_cam_image.shape, im.shape)
     final_image = np.hstack((image, detections_img, renormalized_cam_image, im))
     caption = "Results using YOLOv5"
     return Image.fromarray(final_image), caption