Spaces:

olive100
/

detect

Runtime error

olive100 commited on Jul 25, 2023

Commit

f349298

•

1 Parent(s): 2aee13b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from torchkeras import plots
 model = ultralytics.YOLO('yolov8n.pt')
 #load class_names
-yaml_path = str(Path(ultralytics.__file__).parent/'datasets/coco128.yaml')
 class_names = utils.yaml_load(yaml_path)['names']
 def detect(img):

 model = ultralytics.YOLO('yolov8n.pt')
 #load class_names
+yaml_path = str('coco128.yaml')
 class_names = utils.yaml_load(yaml_path)['names']
 def detect(img):