Fix: Gallery return issue.

app.py

@@ -59,6 +59,7 @@ def load_yolo_model(version="yolov5"):
     model.cpu()
     return model
 
+# Main function for Grad-CAM visualization
 # Main function for Grad-CAM visualization
 def process_image(image, yolo_versions=["yolov5"]):
     image = np.array(image)
@@ -70,7 +71,7 @@ def process_image(image, yolo_versions=["yolov5"]):
     transform = transforms.ToTensor()
     tensor = transform(img_float).unsqueeze(0)
 
-    # Initialize list to store result images
+    # Initialize list to store result images with captions
     result_images = []
 
     # Process each selected YOLO model
@@ -96,13 +97,13 @@ def process_image(image, yolo_versions=["yolov5"]):
         renormalized_cam = scale_cam_image(renormalized_cam)
         renormalized_cam_image = show_cam_on_image(img_float, renormalized_cam, use_rgb=True)
 
-        # Concatenate images
+        # Concatenate images and prepare the caption
         final_image = np.hstack((rgb_img, cam_image, renormalized_cam_image))
-        result_images.append((yolo_version, Image.fromarray(final_image)))
+        caption = f"Results using {yolo_version}"
+        result_images.append((Image.fromarray(final_image), caption))
 
     return result_images
 
-# Gradio Interface
 interface = gr.Interface(
     fn=process_image,
     inputs=[