.

sorghum_pipeline/pipeline.py

@@ -46,19 +46,17 @@ class SorghumPipeline:
     def run(self, single_image_path: str) -> Dict[str, Any]:
         """Run pipeline on single image."""
         logger.info("Processing single image...")
-        
-        from PIL import Image
         import time
-        
+        import cv2
+
         start = time.perf_counter()
         
-        # Load image
-        img = Image.open(single_image_path)
-        arr = np.array(img)
-        print("Debug: Uploaded image shape: ", arr.shape)
+
+        arr = cv2.imread(single_image_path, cv2.IMREAD_UNCHANGED)
+        print("Debug: Uploaded image shape: ", arr.shape, arr.dtype)  # (1024, 1024), uint16
         plants = {
             "demo": {
-                "raw_image": (img, Path(single_image_path).name),
+                "raw_image": (arr, Path(single_image_path).name),
                 "plant_name": "demo",
             }
         }