Update app.py

app.py

@@ -60,7 +60,7 @@ def yolov8_img_inference(
     object_counts = {x: 0 for x in names}
     for r in results:
         for c in r.boxes.cls:
-            c = c.add(names[int(c)])
+            c = (names[int(c)])
             if c in names:
                 object_counts[c] += 1
             elif c not in names:
@@ -108,23 +108,7 @@ def yolov8_img_inference(
     # output_image = visualize_object_predictions(image=image, object_prediction_list=object_prediction_list)
     # return output_image['image']
     # render = render_result(model=model, image=image, result=results[0])
-def count_objects(predictions, target_classes):
-    object_counts = {x: 0 for x in target_classes}
-    for prediction in predictions:
-        for c in prediction.boxes.cls:
-            c = int(c)
-            if c in target_classes:
-                object_counts[c] += 1
-            elif c not in target_classes:
-                object_counts[c] = 1
-
-    present_objects = object_counts.copy()
-
-    for i in object_counts:
-        if object_counts[i] < 1:
-            present_objects.pop(i)
-
-    return present_objects    
+