AjaykumarPilla commited on
Commit
7a3575c
·
verified ·
1 Parent(s): c62a88a

Update gully_drs_core/model_utils.py

Browse files
Files changed (1) hide show
  1. gully_drs_core/model_utils.py +2 -1
gully_drs_core/model_utils.py CHANGED
@@ -2,5 +2,6 @@
2
  from ultralytics import YOLO
3
 
4
  def load_model():
5
- model = YOLO('yolov5s.pt') # Replace with custom model if available
 
6
  return model
 
2
  from ultralytics import YOLO
3
 
4
  def load_model():
5
+ # Load your custom YOLOv8 model
6
+ model = YOLO('best.pt') # 👈 Your trained cricket model
7
  return model