Update app.py

app.py

@@ -58,14 +58,21 @@ def yolov8_img_inference(
     # count_result = results[0].boxes.cls.tolist()
 
     object_counts = {x: 0 for x in names}
-    for r in results:
-        for c in r.boxes.cls:
-            c = (names[int(c)])
-            if c in names:
-                object_counts[c] += 1
-            elif c not in names:
-                object_counts[c] = 1
-
+    # for r in results:
+    #     for c in r.boxes.cls:
+    #         c = (names[int(c)])
+    #         if c in names:
+    #             object_counts[c] += 1
+    #         elif c not in names:
+    #             object_counts[c] = 1
+    clist = results[0].boxes.cls
+    cls = set()
+    for c in clist:
+        cls = cls.add(names[int(c)])
+        if c in names:
+            object_counts[c] += 1
+        elif c not in names:
+            object_counts[c] = 1
     present_objects = object_counts.copy()
 
     for i in object_counts: