api change 4

main.py

@@ -9,6 +9,7 @@ app = FastAPI(
     version="1.0.0"
 )
 
+# Load model & class names
 try:
     model, target_names = joblib.load("iris_knn.pkl")
 except:
@@ -38,7 +39,14 @@ def predict_iris(data: IrisData):
     ]])
 
     pred = model.predict(arr)[0]
+    proba = model.predict_proba(arr)[0]
+
+    probability_dict = {
+        str(target_names[i]): float(proba[i]) for i in range(len(target_names))
+    }
+
     return {
         "input": data.dict(),
-        "predicted_class": str(target_names[pred])
+        "predicted_class": str(target_names[pred]),
+        "class_probabilities": probability_dict
     }