Update app.py

app.py

@@ -56,14 +56,18 @@ def yolov8_img_inference(
     # count 'car' objects in the results
     # count_result = results[0].boxes.cls[0].item()
     # count_result = results[0].boxes.cls.tolist()
+
+    for r in results:
+        for c in r.boxes.cls:
+            count = (names[int(c)])
     
-    clist= results[0].boxes.cls.tolist()
-    cls = set()
+    # 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, cls
+    return render, count
     
     # results = model.predict(image, imgsz=image_size, return_outputs=True)
     # results = model.predict(image)