added class probabilities

app.py

@@ -1,6 +1,7 @@
 # %%
 import gradio as gr
 import joblib
+import numpy as np
 loaded_rf_2way = joblib.load("STPI_2WAY_RandomForest.joblib")
 loaded_rf_3way = joblib.load("STPI_3WAY_RandomForest.joblib")
 
@@ -38,17 +39,26 @@ Abs_Diff_t_0_5_MaxValue,Abs_Diff_t_1_0_MaxValue,Abs_Diff_t_2_0_MaxValue,Optional
     result_2way = loaded_rf_2way.predict([X])
     print('The patient is ', outcome_decoded[int(result_2way)], ' through the 2way method')
 
+    probs_2way = loaded_rf_2way.predict_proba([X])
+    probs_2way = str(np.round(probs_2way[0], decimals=2))
+    print('2 way class Probabilities (Normal/KC) are ', probs_2way)
+
     result_3way = loaded_rf_3way.predict([X])
+
+    probs_3way = loaded_rf_3way.predict_proba([X])
+    probs_3way = str(np.round(probs_3way[0], decimals=2))
+    print('3 way class Probabilities (Normal/Suspect/KC) are ', probs_3way)
+
     if result_2way == 0:
         print('The patient is ', outcome_decoded[int(result_3way)], 'through the 3way method')
         # result = 'The 3-way classification resulted in a ', outcome_decoded[int(result_3way)] + ' patient.'
         # further_analysis = 'Futher analysis using the 2-way classification resulted in a ' + outcome_decoded[int(result_2way)] + ' label.'
-        return 'The 3-way classification resulted in a ' + outcome_decoded[int(result_3way)] + ' patient. Futher analysis using the 2-way classification resulted in a ' + outcome_decoded[int(result_2way)] + ' label.'
+        return 'The 3-way classification resulted in a ' + outcome_decoded[int(result_3way)] + ' patient. Futher analysis using the 2-way classification resulted in a ' + outcome_decoded[int(result_2way)] + ' label. ' + '2 way class Probabilities (Normal/KC) are ' + probs_2way + ' and 3 way class Probabilities (Normal/Suspect/KC) are ' + probs_3way
 
     # result = 'The 2-way classification resulted in a ', outcome_decoded[int(result_2way)] + ' patient.'
     # further_analysis = 'Futher analysis using the 3-way classification resulted in a ' + outcome_decoded[int(result_3way)] + ' label.'
 
-    return 'The 2-way classification resulted in a ' + outcome_decoded[int(result_2way)] + ' patient. Futher analysis using the 3-way classification resulted in a ' + outcome_decoded[int(result_3way)] + ' label.'
+    return 'The 2-way classification resulted in a ' + outcome_decoded[int(result_2way)] + ' patient. Futher analysis using the 3-way classification resulted in a ' + outcome_decoded[int(result_3way)] + ' label. ' + '2 way class Probabilities (Normal/KC) are ' + probs_2way + ' and 3 way class Probabilities (Normal/Suspect/KC) are ' + probs_3way
 
 iface = gr.Interface(
     fn=STPI, 
@@ -58,5 +68,7 @@ iface = gr.Interface(
     # "number",
     "number", "number","number","text"],
     outputs="text")
-iface.launch(share=True)
+iface.launch(
+    # share=True
+    )
 # %%