Update inference_utils.py

inference_utils.py

@@ -8,7 +8,9 @@ from paddleocr import PaddleOCR
 from difflib import get_close_matches
 
 from huggingface_hub import hf_hub_download
-
+from torch.serialization import safe_globals
+from ultralytics.nn.tasks import DetectionModel
+from ultralytics import YOLO
 # Download to local cache
 
 
@@ -18,11 +20,15 @@ def load_models():
     # vehicle_classifier = YOLO("https://huggingface.co/Prabhat51/number-plate-models/blob/main/veh_class.pt")
     # plate_detector = YOLO("https://huggingface.co/Prabhat51/number-plate-models/blob/main/plate_detect.pt")
     veh_detect_path = hf_hub_download(repo_id="Prabhat51/number-plate-models", filename="veh_detect.pt")
-    vehicle_detector = YOLO(veh_detect_path)
+    with safe_globals([DetectionModel]):
+        vehicle_detector = YOLO(veh_detect_path)
+    # vehicle_detector = YOLO(veh_detect_path)
     vehicle_classifier_path = hf_hub_download(repo_id="Prabhat51/number-plate-models", filename="veh_class.pt")
     vehicle_classifier = YOLO(vehicle_classifier_path)
     plate_detector_path = hf_hub_download(repo_id="Prabhat51/number-plate-models", filename="plate_detect.pt")
-    plate_detector = YOLO(plate_detector_path)
+    with safe_globals([DetectionModel]):
+        vehicle_detector = YOLO(plate_detector_path)
+    # plate_detector = YOLO(plate_detector_path)
     ocr_reader = PaddleOCR(use_angle_cls=True, lang='en')
     return vehicle_detector, vehicle_classifier, plate_detector, ocr_reader