Updated results dict to include masks

app.py

@@ -197,8 +197,15 @@ async def predict_pipeline_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: