Update README.md

README.md

@@ -47,24 +47,23 @@ for chunk in chunks:
     result = pipe(chunk)
     results.append(result)
 
-# Hàm để tìm kết quả có điểm số cao nhất
-def get_highest_score(result_list):
-    highest_score = -1
-    highest_result = None
-    for result in result_list:
-        score = result[0]['score']
-        if score > highest_score:
-            highest_score = score
-            highest_result = result
-    return highest_result
-
-# call funtion get_highest_score
-highest_result = get_highest_score(results)
+# Function to get most common label
+def get_most_common_label(results_list):
+    label_counts = {}
+    for result in results_list:
+        label = result[0]['label']
+        label_counts[label] = label_counts.get(label, 0) + 1
+
+    most_common_label = max(label_counts, key=label_counts.get)
+    return most_common_label
+
+# call funtion get_most_common_label
+most_common_label = get_most_common_label(results)
 
 # In ra nhãn và điểm số của kết quả có điểm số cao nhất
 if highest_result:
-    label = highest_result[0]['label']
-    score = highest_result[0]['score']
+    label = most_common_label[0]['label']
+    score = most_common_label[0]['score']
     print("(Label):", label)
     print("(Score):", score)
 else: