Update app.py

app.py

@@ -60,7 +60,7 @@ def yolo_predict(image: np.ndarray, iou_thresh: float = 0.75, thresh: float = 0.
     )
     plot_img = draw_predictions(image, nms_boxes, class_labels=config.PASCAL_CLASSES)
     
-    return plot_img
+    return [plot_img]
 
 
 def draw_predictions(image: np.ndarray, boxes: list[list], class_labels: list[str]) -> np.ndarray:
@@ -118,7 +118,7 @@ demo = gr.Interface(
         gr.Slider(0, 1, value=0.75, step=0.05, label="IOU Threshold"),
         gr.Slider(0, 1, value=0.75, step=0.05, label="Threshold")
     ],
-    outputs=gr.Image(label="Prediction"),
+    outputs=gr.Gallery(rows=1, columns=1),
     examples=[
        ["examples/000001.jpg", 0.75, 0.75],
        ["examples/000002.jpg", 0.75, 0.75],