Update working_yolo_pipeline.py

working_yolo_pipeline.py

@@ -3,16 +3,30 @@ import fitz  # PyMuPDF
 import numpy as np
 import cv2
 
+# import torch
+# from torch.nn.modules.conv import Conv2d
+# from ultralytics.nn.tasks import DetectionModel
+# from torch.nn.modules.container import Sequential
+
+# torch.serialization.add_safe_globals([
+#     Conv2d,         # The exact unsupported global shown in your log
+#     DetectionModel,
+#     Sequential,
+# ])
+
+
 import torch
-from torch.nn.modules.conv import Conv2d
-from ultralytics.nn.tasks import DetectionModel
-from torch.nn.modules.container import Sequential
-
-torch.serialization.add_safe_globals([
-    Conv2d,         # The exact unsupported global shown in your log
-    DetectionModel,
-    Sequential,
-])
+import torch.serialization
+
+_original_torch_load = torch.load
+
+def patched_torch_load(*args, **kwargs):
+    # FORCE classic behavior
+    kwargs["weights_only"] = False
+    return _original_torch_load(*args, **kwargs)
+
+torch.load = patched_torch_load
+
 
 
 import json