update probability method

main.py

@@ -7,11 +7,6 @@ import io
 import pandas as pd
 import os
 
-
-# Matiin log kalo run di Windows doang
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
-os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
-
 app = FastAPI()
 
 
@@ -96,9 +91,14 @@ async def predict(file: UploadFile = File(...)):
     else:
         predicted_label = "Tidak diketahui"
 
-    probability = {}
+    probability = []
     for i, prob in enumerate(preds[0]):
-        probability[CLASS_NAMES[i]] = round(float(prob) * 100, 2)
+        probability.append({
+            "label": CLASS_NAMES[i],
+            "score": round(float(prob) * 100, 2)
+        })
+
+    probability = sorted(probability, key=lambda x: x["score"], reverse=True)
 
     data_nutrisi = get_nutrition_data(predicted_label)