Spaces:
Runtime error
Runtime error
Update models/detection/detector.py
Browse files
models/detection/detector.py
CHANGED
|
@@ -54,8 +54,10 @@ class ObjectDetector:
|
|
| 54 |
cache_dir="models/detection/weights",
|
| 55 |
force_download=True #Clear cache
|
| 56 |
)
|
|
|
|
| 57 |
|
| 58 |
self.device = device
|
|
|
|
| 59 |
self.model = YOLO(weights_path)
|
| 60 |
print("Model object type:", type(self.model))
|
| 61 |
print("Model class string:", self.model.__class__)
|
|
|
|
| 54 |
cache_dir="models/detection/weights",
|
| 55 |
force_download=True #Clear cache
|
| 56 |
)
|
| 57 |
+
|
| 58 |
|
| 59 |
self.device = device
|
| 60 |
+
print("Loading weights from:", weights_path)
|
| 61 |
self.model = YOLO(weights_path)
|
| 62 |
print("Model object type:", type(self.model))
|
| 63 |
print("Model class string:", self.model.__class__)
|