Spaces:
Runtime error
Runtime error
Commit
·
ebd31f9
1
Parent(s):
8967683
Update detection.py
Browse files- detection.py +1 -1
detection.py
CHANGED
@@ -3,7 +3,7 @@ import IPython
|
|
3 |
from PIL import ImageColor
|
4 |
from ultralytics import YOLO
|
5 |
|
6 |
-
class
|
7 |
def __init__(self, model_name='Yolov8'):
|
8 |
self.model_name = model_name
|
9 |
self.model = self.load_model()
|
|
|
3 |
from PIL import ImageColor
|
4 |
from ultralytics import YOLO
|
5 |
|
6 |
+
class ObjectDetection:
|
7 |
def __init__(self, model_name='Yolov8'):
|
8 |
self.model_name = model_name
|
9 |
self.model = self.load_model()
|