Update app.py

app.py

@@ -58,21 +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
-    clist = results[0].boxes.cls
-    cls = set()
-    for cno in clist:
-        cls.add(model.names[int(cno)])
-        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 = 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 cno in clist:
+    #     cls.add(model.names[int(cno)])
+    #     if cno in names:
+    #         object_counts[cno] += 1
+    #     elif cno not in names:
+    #         object_counts[cno] = 1
     present_objects = object_counts.copy()
 
     for i in object_counts: