Update app.py

app.py

@@ -57,6 +57,20 @@ def yolov8_img_inference(
     # count_result = results[0].boxes.cls[0].item()
     # count_result = results[0].boxes.cls.tolist()
 
+    object_counts = {x: 0 for x in names}
+    for r in results:
+        for c in results.boxes.cls:
+            c = 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:
+        if object_counts[i] < 1:
+            present_objects.pop(i)
     
     # clist= results[0].boxes.cls.tolist()
     # cls = set()
@@ -64,7 +78,7 @@ def yolov8_img_inference(
     #     cls.add(model.names[int(cno)])
         
     # count_result = results.pandas().xyxy[0].value_counts('name')
-    return render
+    return render, present_objects
     # results = model.predict(image, imgsz=image_size, return_outputs=True)
     # results = model.predict(image)
     # object_prediction_list = []