RishitJavia commited on
Commit
a62a4b8
·
1 Parent(s): ec24258

Fix model dependency.

Browse files
Files changed (1) hide show
  1. src/crop_video.py +1 -1
src/crop_video.py CHANGED
@@ -45,7 +45,7 @@ class YOLOCrop(CropVideo):
45
 
46
  def __init__(self, method=None, model_path=None):
47
  super().__init__('yolo')
48
- self.model_path = model_path or '../data/models/yolo/yolov5x.pt'
49
  self.load_model(self.model_path)
50
 
51
  def load_model(self, model_path):
 
45
 
46
  def __init__(self, method=None, model_path=None):
47
  super().__init__('yolo')
48
+ self.model_path = model_path or 'models/yolo/yolov5x.pt'
49
  self.load_model(self.model_path)
50
 
51
  def load_model(self, model_path):