Hugging Face's logo

Spaces:
Prabhat51
/
EdgeAI_Project
Runtime error

App Files Community
Prabhat51 commited on
Commit
59c285b
·
verified ·
1 Parent(s): 952c05b

Update inference_utils.py

Browse files
Files changed (1) hide show
  1. inference_utils.py +3 -3
inference_utils.py CHANGED
@@ -9,9 +9,9 @@ from difflib import get_close_matches
9
 
10
  # Load models from Hugging Face
11
  def load_models():
12
- vehicle_detector = YOLO("https://huggingface.co/Prabhat51/veh-detect/resolve/main/veh_detect.pt")
13
- vehicle_classifier = YOLO("https://huggingface.co/Prabhat51/veh-class/resolve/main/veh_class.pt")
14
- plate_detector = YOLO("https://huggingface.co/Prabhat51/plate-detect/resolve/main/plate_detect.pt")
15
  ocr_reader = PaddleOCR(use_angle_cls=True, lang='en')
16
  return vehicle_detector, vehicle_classifier, plate_detector, ocr_reader
17
 
 
9
 
10
  # Load models from Hugging Face
11
  def load_models():
12
+ vehicle_detector = YOLO("https://huggingface.co/Prabhat51/number-plate-models/blob/main/veh_detect.pt")
13
+ vehicle_classifier = YOLO("https://huggingface.co/Prabhat51/number-plate-models/blob/main/veh_class.pt")
14
+ plate_detector = YOLO("https://huggingface.co/Prabhat51/number-plate-models/blob/main/plate_detect.pt")
15
  ocr_reader = PaddleOCR(use_angle_cls=True, lang='en')
16
  return vehicle_detector, vehicle_classifier, plate_detector, ocr_reader
17