linhcuem commited on
Commit
1fc60a5
·
1 Parent(s): 1e4a68a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def yolov8_img_inference(
63
  # cls.add(model.names[int(cno)])
64
 
65
  # count_result = results.pandas().xyxy[0].value_counts('name')
66
- return render, results[0]
67
 
68
  # results = model.predict(image, imgsz=image_size, return_outputs=True)
69
  # results = model.predict(image)
 
63
  # cls.add(model.names[int(cno)])
64
 
65
  # count_result = results.pandas().xyxy[0].value_counts('name')
66
+ return render, results[0].boxes.cls
67
 
68
  # results = model.predict(image, imgsz=image_size, return_outputs=True)
69
  # results = model.predict(image)