linhcuem commited on
Commit
5bbbccb
·
1 Parent(s): a1a4ce9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def yolov8_img_inference(
55
  # obj_id = list(names)[list(names.values()).index([0])]
56
  # count 'car' objects in the results
57
  # count_result = results[0].boxes.cls[0].item()
58
- count_result = results[0].boxes.cls.tolist()
59
 
60
  # count_result = results.pandas().xyxy[0].value_counts('name')
61
  return render, count_result
 
55
  # obj_id = list(names)[list(names.values()).index([0])]
56
  # count 'car' objects in the results
57
  # count_result = results[0].boxes.cls[0].item()
58
+ count_result = results[0].boxes.cls.tolist().pandas()
59
 
60
  # count_result = results.pandas().xyxy[0].value_counts('name')
61
  return render, count_result