linhcuem commited on
Commit
0e69d8b
1 Parent(s): 0bf4aff

Update app.py

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