Update working_yolo_pipeline.py

working_yolo_pipeline.py

@@ -1,19 +1,31 @@
+
+import fitz  # PyMuPDF
+import numpy as np
+import cv2
+
+
+import torch
+from ultralytics.nn.modules.conv import Conv
+from ultralytics.nn.tasks import DetectionModel
+from torch.nn.modules.container import Sequential
+
+torch.serialization.add_safe_globals([
+    DetectionModel,
+    Sequential,
+    Conv,
+])
+
+
 import json
 import argparse
 import os
 import re
-import torch
+
 import torch.nn as nn
 from TorchCRF import CRF
 from transformers import LayoutLMv3TokenizerFast, LayoutLMv3Model, LayoutLMv3Config
 from typing import List, Dict, Any, Optional, Union, Tuple
-import fitz  # PyMuPDF
-import numpy as np
-import cv2
 from ultralytics import YOLO
-from ultralytics.nn.tasks import DetectionModel
-from ultralytics.nn.modules.conv import Conv as UltralyticsConv
-from torch.nn.modules.container import Sequential
 import glob
 import pytesseract
 from PIL import Image
@@ -212,12 +224,6 @@ def get_latex_from_base64(base64_string: str) -> str:
 # --- CONFIGURATION AND CONSTANTS ---
 # ============================================================================
 
-torch.serialization.add_safe_globals([
-    DetectionModel,
-    Sequential,
-    UltralyticsConv,
-])
-
 
 
 # NOTE: Update these paths to match your environment before running!