.

sorghum_pipeline/pipeline.py

@@ -50,16 +50,26 @@ class SorghumPipeline:
         import time
         
         start = time.perf_counter()
-        
-        # Load image
+        import imghdr
         import tifffile
+        import cv2
+        from PIL import Image
+
+        kind = imghdr.what(single_image_path)
 
-        arr = tifffile.imread(single_image_path)
-        print("DEBUG tifffile:", arr.shape, arr.dtype)  
+        if kind == "tiff":
+            arr = tifffile.imread(single_image_path)
+            print("DEBUG loaded TIFF:", arr.shape, arr.dtype)
+            img = Image.fromarray(arr)  # keep pipeline compatibility
+        else:
+            arr = cv2.imread(single_image_path, cv2.IMREAD_UNCHANGED)
+            print("DEBUG loaded non-TIFF:", arr.shape, arr.dtype)
+            img = Image.fromarray(arr)
+                
         print("Debug: Uploaded image shape: ", arr.shape)
         plants = {
             "demo": {
-                "raw_image": (arr, Path(single_image_path).name),
+                "raw_image": (img, Path(single_image_path).name),
                 "plant_name": "demo",
             }
         }