Spaces:
Runtime error
Runtime error
Commit
·
a62a4b8
1
Parent(s):
ec24258
Fix model dependency.
Browse files- 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 '
|
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):
|