Update app.py

app.py

@@ -56,6 +56,9 @@ def yolov8_inference(
     model.overrides["conf"] = conf_threshold
     model.overrides["iou"] = iou_threshold
     results = model.predict(image, imgsz=image_size, return_outputs=True)
+    masks=[]
+    for i in results:
+        masks.append([i.masks,i.labels])
     renders = []
     for image_results in model.predict(image, imgsz=image_size, return_outputs=True):
         render = render_model_output(
@@ -63,7 +66,7 @@ def yolov8_inference(
         )
         renders.append(render)
 
-    return renders[0]
+    return masks
 
 
 inputs = [