Update working_yolo_pipeline.py

torch.serialization.add_safe_globals([DetectionModel])

from ultralytics.nn.tasks import DetectionModel

working_yolo_pipeline.py

@@ -11,6 +11,7 @@ import fitz  # PyMuPDF
 import numpy as np
 import cv2
 from ultralytics import YOLO
+from ultralytics.nn.tasks import DetectionModel
 import glob
 import pytesseract
 from PIL import Image
@@ -202,6 +203,9 @@ def get_latex_from_base64(base64_string: str) -> str:
 # --- CONFIGURATION AND CONSTANTS ---
 # ============================================================================
 
+torch.serialization.add_safe_globals([DetectionModel])
+
+
 # NOTE: Update these paths to match your environment before running!
 WEIGHTS_PATH = 'best.pt'
 DEFAULT_LAYOUTLMV3_MODEL_PATH = "98.pth"