Spaces:

foduucom
/

plant-leaf-detection-classification-yolov8

Runtime error

nehulagrawal commited on Jul 28, 2023

Commit

098a57d

1 Parent(s): 7e18357

Update detection.py

Files changed (1) hide show

detection.py CHANGED Viewed

@@ -4,7 +4,7 @@ from PIL import ImageColor
 from ultralytics import YOLO
 class ObjectDetection:
-    def __init__(self, model_name='yolov8'):
         self.model_name = model_name
         self.model = self.load_model()
         self.classes = self.model.names

 from ultralytics import YOLO
 class ObjectDetection:
+    def __init__(self, model_name='Yolov8'):
         self.model_name = model_name
         self.model = self.load_model()
         self.classes = self.model.names