Update api_server.py

api_server.py

@@ -114,7 +114,7 @@ def predict():
         labels = result.boxes.cls  # Get predicted label IDs
         label_names = [model.names[int(label)] for label in labels]  # Convert to names
 
-        element_counts = Counter(my_list)
+        element_counts = Counter(label_names)
 
         encoded_images=[]
         element_list =[]