stack components

yolov5.py

@@ -77,13 +77,13 @@ def xai_yolov5(image):
     cam_image, renormalized_cam_image = generate_cam_image(model, target_layers, tensor, image, boxes)
 
     rgb_img_float, batch_explanations, result = dff_nmf(image, target_lyr = -5, n_components = 8)
-    #result = np.hstack(result)
+    result = np.hstack(result)
     im = visualize_batch_explanations(rgb_img_float, batch_explanations)
     
     # Combine results
     final_image = np.hstack((image, detections_img, renormalized_cam_image, im))
     caption = "Results using YOLOv5"
-    return Image.fromarray(final_image), caption, result[0]
+    return Image.fromarray(final_image), caption, result