Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr 
 import pandas as pd 
 from skimage import data
-from ultralytics.data import utils 
+from ultralytics.yolo.data import utils 
 import ultralytics 
 from pathlib import Path
 from torchkeras import plots
@@ -9,17 +9,16 @@ from torchkeras import plots
 model = ultralytics.YOLO('yolov8n.pt')
  
 #load class_names
-yaml_path = str('coco128.yaml') 
+yaml_path = str(Path(ultralytics.__file__).parent/'datasets/coco128.yaml') 
 class_names = utils.yaml_load(yaml_path)['names']
 
 def detect(img):
     if isinstance(img,str):
         img = get_url_img(img) if img.startswith('http') else Image.open(img).convert('RGB')
     result = model.predict(source=img)
-    if len(result[0].boxes.data)>0:
-        # vis = plots.plot_detection(img,boxes=result[0].boxes.data,
-        #              class_names=class_names, min_score=0.2)
-        print(result[0].boxes)
+    if len(result[0].boxes.boxes)>0:
+        vis = plots.plot_detection(img,boxes=result[0].boxes.boxes,
+                     class_names=class_names, min_score=0.2)
     else:
         vis = img
     return vis