Update app.py

app.py

@@ -55,7 +55,7 @@ def yolov8_img_inference(
     # obj_id = list(names)[list(names.values()).index([0])]
     # count 'car' objects in the results
     # count_result = results[0].boxes.cls[0].item()
-    count_result = results[0].boxes.cls[0].tolist()
+    count_result = results[0].boxes.cls[].tolist()
   
     # count_result = results.pandas().xyxy[0].value_counts('name')
     return render, count_result