AndySAnker commited on
Commit
dd5679e
1 Parent(s): 5b80ef3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -83,6 +83,7 @@ def POMPredicter(POMFinder, Gr, y_onehotenc_cat):
83
  y_pred_proba = y_pred_proba[:,1];
84
  res = sorted(range(len(y_pred_proba)), key = lambda sub: y_pred_proba[sub]);
85
  res.reverse();
 
86
  name = str(y_onehotenc_cat.categories_[0][res[0]])[2:-2]
87
  if name.startswith("icsd"):
88
  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 {100*y_pred_proba[res[0]]:.2f} %</span> <hr/>',unsafe_allow_html=True)
 
83
  y_pred_proba = y_pred_proba[:,1];
84
  res = sorted(range(len(y_pred_proba)), key = lambda sub: y_pred_proba[sub]);
85
  res.reverse();
86
+ st.write(str(y_onehotenc_cat.categories_[0][res[0]])[2:-2])
87
  name = str(y_onehotenc_cat.categories_[0][res[0]])[2:-2]
88
  if name.startswith("icsd"):
89
  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 {100*y_pred_proba[res[0]]:.2f} %</span> <hr/>',unsafe_allow_html=True)