Update app.py

app.py

@@ -49,12 +49,13 @@ def yolov8_img_inference(
     render = render_result(model=model, image=image, result=results[0])
     # get the model names list
     names = model.names
-    # get the 'car' class id
-    # obj_id = list(names)[list(names.values()).index('hop_dln','hop_jn','hop_vtg','hop_ytv','lo_kids', 'lo_ytv','loc_dln','loc_jn','loc_kids','loc_ytv')]
+    # get the 'obj' class id
+    obj_id = list(names)[list(names.values()).index('hop_dln')
+    # ,'hop_jn','hop_vtg','hop_ytv','lo_kids', 'lo_ytv','loc_dln','loc_jn','loc_kids','loc_ytv')]
     # 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.tolist().value_counts()
+    count_result = results[0].boxes.cls.tolist().count(obj_id)
     # count_result = results.pandas().xyxy[0].value_counts('name')
     return render, names