Update app.py

app.py

@@ -78,6 +78,9 @@ def POMPredicter(POMFinder, Gr, y_onehotenc_cat):
     y_pred_proba = y_pred_proba[:,1];
     res = sorted(range(len(y_pred_proba)), key = lambda sub: y_pred_proba[sub]);
     res.reverse();
+    st.write(y_onehotenc_cat.categories_)
+    st.write(y_onehotenc_cat.categories_[0])
+    st.write(y_onehotenc_cat.categories_[0][res[0]])
     st.markdown(f'<span style="font-size: 24px; color: green;">The 1st guess from the model is: <b>{str(y_onehotenc_cat.categories_[0][res[0]])[2:-2]+"cale.xyz"}</b> with a probability of {y_pred_proba[res[0]]:.2f} %</span> <hr/>',unsafe_allow_html=True)
     st.markdown(f'<span style="font-size: 24px; color: green;">The 2nd guess from the model is: <b>{str(y_onehotenc_cat.categories_[0][res[1]])[2:-2]+"cale.xyz"}</b> with a probability of {y_pred_proba[res[1]]:.2f} %</span> <hr/>',unsafe_allow_html=True)
     st.markdown(f'<span style="font-size: 24px; color: green;">The 3rd guess from the model is: <b>{str(y_onehotenc_cat.categories_[0][res[2]])[2:-2]+"cale.xyz"}</b> with a probability of {y_pred_proba[res[2]]:.2f} %</span> <hr/>',unsafe_allow_html=True)