Update my_model/object_detection.py
Browse files
my_model/object_detection.py
CHANGED
|
@@ -80,7 +80,7 @@ class ObjectDetector:
|
|
| 80 |
"""
|
| 81 |
|
| 82 |
try:
|
| 83 |
-
model_path =
|
| 84 |
if model_path and os.path.exists(model_path):
|
| 85 |
self.model = torch.hub.load(model_path, model_version, pretrained=pretrained, source='local')
|
| 86 |
else:
|
|
|
|
| 80 |
"""
|
| 81 |
|
| 82 |
try:
|
| 83 |
+
model_path = get_model_path ('yolov5')
|
| 84 |
if model_path and os.path.exists(model_path):
|
| 85 |
self.model = torch.hub.load(model_path, model_version, pretrained=pretrained, source='local')
|
| 86 |
else:
|