Update run_inference.py

run_inference.py

@@ -2,6 +2,13 @@ import sys, os
 import cv2
 import torch
 
+# Monkey-patch torch.load to disable weights_only safety for trusted checkpoint
+_orig_torch_load = torch.load
+torch.load = lambda f, **kwargs: _orig_torch_load(f, weights_only=False, **kwargs)
+
+import cv2
+import torch
+
 # Allow unpickling of HistoryBuffer
 from mmengine.logging.history_buffer import HistoryBuffer
 # Allow numpy reconstruct, ndarray, dtype, Float64DType, Int64DType, and getattr
@@ -10,6 +17,8 @@ import numpy.core.multiarray as cam
 import numpy.dtypes as dtypes
 
 torch.serialization.add_safe_globals([
+    cam.scalar,  # allow numpy scalar types
+
     HistoryBuffer,
     cam._reconstruct,
     np.ndarray,