Update app.py

app.py

@@ -16,13 +16,10 @@ def yolo_inference(images, model_id, conf_threshold, iou_threshold, max_detectio
         show_conf=True,
     )
 
-    annotated_images = []
-    for result in results:
-        im_array = result.plot()
-        im = Image.fromarray(im_array)
-        annotated_images.append(im)
-    
-    return annotated_images  
+    for r in results:
+        image_array = r.plot()
+        image = Image.fromarray(image_array[..., ::-1])
+    return image
 
 interface = gr.Interface(
     fn=yolo_inference,