Spaces:
Running
on
Zero
Running
on
Zero
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__)
|