linhcuem commited on
Commit
2fb56ea
·
1 Parent(s): 3dfb48a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ def yolov8_img_inference(
59
  clist= results[0].boxes.cls
60
  cls = set()
61
  for cno in clist:
62
- cls.add(model.names[int(cno)])
63
 
64
  # count_result = results.pandas().xyxy[0].value_counts('name')
65
  return render, cls
 
59
  clist= results[0].boxes.cls
60
  cls = set()
61
  for cno in clist:
62
+ cls.add(model.names[int(cno)]).count()
63
 
64
  # count_result = results.pandas().xyxy[0].value_counts('name')
65
  return render, cls