Change YOLOS output from image to gallery of images

app.py

@@ -66,7 +66,7 @@ def detect_objects(img: Image.Image):
         )
         box_images.append(img.crop((box[0], box[1], box[2], box[3])))
 
-    return box_images[0]
+    return box_images
 
 
 if __name__ == "__main__":
@@ -82,7 +82,7 @@ if __name__ == "__main__":
         with gr.Tab("Object Detection"):
             with gr.Row():
                 yolos_input = gr.Image(type="pil")
-                yolos_output = gr.Image()
+                yolos_output = gr.Gallery(label="Detected Objects", show_label=True)
             yolos_button = gr.Button("Submit")
 
         with gr.Tab("Inference"):