Update app.py

app.py

@@ -131,9 +131,15 @@ async def predict_yolo_api(file: UploadFile = File(...)):
                 "classes": results.boxes.cls.tolist() if results.boxes is not None else None,
                 "names": results.names,
                 "orig_shape": results.orig_shape,
-                "speed": results.speed
+                "speed": results.speed,
+                "masks": {
+                    "data": results.masks.data.tolist() if results.masks is not None else None,
+                    "orig_shape": results.masks.orig_shape if results.masks is not None else None,
+                    "xy": [seg.tolist() for seg in results.masks.xy] if results.masks is not None else None,
+                    "xyn": [seg.tolist() for seg in results.masks.xyn] if results.masks is not None else None
+                }
             }
-            
+
             # Handle growth stages if present in the results
             if hasattr(results, 'boxes') and results.boxes is not None:
                 if hasattr(results.boxes, 'data') and len(results.boxes.data) > 0: