Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -61,8 +61,8 @@ def yolov8_img_inference(
|
|
61 |
# for cno in clist:
|
62 |
# cls.add(model.names[int(cno)])
|
63 |
for r in results:
|
64 |
-
|
65 |
-
|
66 |
# count_result = results.pandas().xyxy[0].value_counts('name')
|
67 |
return render, count
|
68 |
|
|
|
61 |
# for cno in clist:
|
62 |
# cls.add(model.names[int(cno)])
|
63 |
for r in results:
|
64 |
+
for c in r.boxes.cls:
|
65 |
+
count = names[int(c)]
|
66 |
# count_result = results.pandas().xyxy[0].value_counts('name')
|
67 |
return render, count
|
68 |
|