Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ def yolov8_img_inference(
|
|
59 |
|
60 |
object_counts = {x: 0 for x in names}
|
61 |
for r in results:
|
62 |
-
for c in
|
63 |
c = int(c)
|
64 |
if c in names:
|
65 |
object_counts[c] += 1
|
|
|
59 |
|
60 |
object_counts = {x: 0 for x in names}
|
61 |
for r in results:
|
62 |
+
for c in r.boxes.cls:
|
63 |
c = int(c)
|
64 |
if c in names:
|
65 |
object_counts[c] += 1
|