Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|