Update working_yolo_pipeline.py

working_yolo_pipeline.py

@@ -1539,7 +1539,8 @@ def preprocess_and_ocr_page(original_img: np.ndarray, model, pdf_path: str,
     # --- STEP 1: YOLO DETECTION ---
     # ====================================================================
     start_time_yolo = time.time()
-    results = model.predict(source=original_img, conf=CONF_THRESHOLD, imgsz=640, verbose=False)
+    # results = model.predict(source=original_img, conf=CONF_THRESHOLD, imgsz=640, verbose=False)
+    results = model.predict(source=original_img, conf=CONF_THRESHOLD, verbose=False)
 
     relevant_detections = []
     if results and results[0].boxes: