prathampatel1 commited on
Commit
402a941
·
verified ·
1 Parent(s): 6b1c23f

Update classify.py

Browse files
Files changed (1) hide show
  1. classify.py +1 -1
classify.py CHANGED
@@ -3,7 +3,7 @@ from ultralytics import YOLO
3
 
4
  class Classification:
5
  def __init__(self) -> None:
6
- self.__cls_model = YOLO('/home/pratham/MDD/bin/model/cls_best.pt')
7
 
8
  def classify_defect(self, image_path) -> pd.DataFrame:
9
  result_cls = self.__cls_model.predict(image_path, stream=False)
 
3
 
4
  class Classification:
5
  def __init__(self) -> None:
6
+ self.__cls_model = YOLO('model/cls_best.pt')
7
 
8
  def classify_defect(self, image_path) -> pd.DataFrame:
9
  result_cls = self.__cls_model.predict(image_path, stream=False)