Update app.py

app.py

@@ -65,28 +65,15 @@ def yolov8_img_inference(
                 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:
         if object_counts[i] < 1:
             present_objects.pop(i)
     
-    # clist= results[0].boxes.cls.tolist()
-    # cls = set()
-    # for cno in clist:
-    #     cls.add(model.names[int(cno)])
-        
-    # count_result = results.pandas().xyxy[0].value_counts('name')
-    # return render, present_objects
-    return render, {names[k]: v for k, v in present_objects.items()}
+
+    return render, {names[k]: v for k, v in present_objects}
     # results = model.predict(image, imgsz=image_size, return_outputs=True)
     # results = model.predict(image)
     # object_prediction_list = []