Update app.py

app.py

@@ -56,13 +56,13 @@ def yolov8_img_inference(
     # count 'car' objects in the results
     # count_result = results[0].boxes.cls[0].item()
     # count_result = results[0].boxes.cls.tolist()
-    # clist= results[0].boxes.cls
-    # cls = set()
-    # for cno in clist:
-    #     cls.add(model.names[int(cno)])
-    for r in results:
-        for c in r.boxes.cls:
-            count = names[int(c)]
+    clist= results[0].boxes.cls.tolist()
+    cls = set()
+    for cno in clist:
+        cls.add(model.names[int(cno)])
+    # for r in results:
+    #     for c in r.boxes.cls:
+    #         count = names[int(c)]
     # count_result = results.pandas().xyxy[0].value_counts('name')
     return render, count