Update inference_utils.py

inference_utils.py

@@ -7,11 +7,22 @@ from datetime import datetime
 from paddleocr import PaddleOCR
 from difflib import get_close_matches
 
+from huggingface_hub import hf_hub_download
+
+# Download to local cache
+
+
 # Load models from Hugging Face
 def load_models():
-    vehicle_detector = YOLO("https://huggingface.co/Prabhat51/number-plate-models/blob/main/veh_detect.pt")
-    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")
+    # vehicle_detector = YOLO("https://huggingface.co/Prabhat51/number-plate-models/blob/main/veh_detect.pt")
+    # 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)
+    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)
     ocr_reader = PaddleOCR(use_angle_cls=True, lang='en')
     return vehicle_detector, vehicle_classifier, plate_detector, ocr_reader