Update app.py

app.py

@@ -132,12 +132,12 @@ async def predict_yolo_api(file: UploadFile = File(...)):
                 "names": results.names,
                 "orig_shape": results.orig_shape,
                 "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
-                }
+                # "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
@@ -204,12 +204,12 @@ async def predict_pipeline_api(file: UploadFile = File(...)):
                 "names": results.names,
                 "orig_shape": results.orig_shape,
                 "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
-                }
+                # "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
+                # }
             }