prathampatel1 commited on
Commit
6b1c23f
·
verified ·
1 Parent(s): 972c1d0

Update detect.py

Browse files
Files changed (1) hide show
  1. detect.py +1 -1
detect.py CHANGED
@@ -3,7 +3,7 @@ from ultralytics import YOLO
3
 
4
  class Detection:
5
  def __init__(self) -> None:
6
- self.__obd_model = YOLO('/home/pratham/MDD/bin/model/obd_best.pt')
7
 
8
  def detect_defect(self, image_path) -> pd.DataFrame:
9
  result_obd = self.__obd_model.predict(image_path, stream=False) # Adjust paths as needed
 
3
 
4
  class Detection:
5
  def __init__(self) -> None:
6
+ self.__obd_model = YOLO('model/obd_best.pt')
7
 
8
  def detect_defect(self, image_path) -> pd.DataFrame:
9
  result_obd = self.__obd_model.predict(image_path, stream=False) # Adjust paths as needed