Update working_yolo_pipeline.py

working_yolo_pipeline.py

@@ -2079,8 +2079,8 @@ def run_document_pipeline(input_path: str, layoutlmv3_model_path: str):
     # 1. INITIALIZE YOLO
     yolo_model = YOLO(WEIGHTS_PATH)
     
-    # 2. DETECT FILE TYPE
-    ext = os.path.splitext(input_path).lower()
+    # 2. DETECT FILE TYPEext = os.path.splitext(input_path)[1].lower()
+    ext = os.path.splitext(input_path)[1].lower()
     is_image = ext in ['.jpg', '.jpeg', '.png', '.bmp', '.tiff', '.webp']
     
     all_pages_data = []