Update app.py

app.py

@@ -43,8 +43,9 @@ def yolov8_img_inference(
     # image = read_image
     results = model.predict(image)
     render = render_result(model=model, image=image, result=results[0])
-    count_result = results.pandas().xyxy[0].value_counts('name')
-    return render 
+    count_result = results[0].boxes.cls.tolist().count('name')
+    # count_result = results.pandas().xyxy[0].value_counts('name')
+    return render, count_result
     
     # results = model.predict(image, imgsz=image_size, return_outputs=True)
     # results = model.predict(image)