Update Home.py

Home.py

@@ -19,11 +19,11 @@ def load_essential_models(scaler_6_path, scaler_8_path, scaler_full_path, clf_6_
 
 def convert_prediction(prediction):
     if (prediction[0] == [0]):
-        return 'S'
+        return 'the small breast group'
     elif (prediction[0] == [1]):
-        return 'M'
+        return 'the medium breast group'
     else:
-        return 'L'
+        return 'the large breast group'
 
 st.markdown("<h1 style='text-align: center;'>Vietnamese Woman Bra Size Classifier</h1>", unsafe_allow_html=True)
 
@@ -171,7 +171,8 @@ if predict:
         y_6 = clf_6.predict(X_6)
         st.success(y_6)
         y_pred_6 = convert_prediction(y_6)
-        styled_text = f"<h3 style='text-align: center;'>We recommend you choosing <span style='color: red; font-weight: bold;'>{y_pred_6}</span> size!</h3>"
+        # styled_text = f"<h3 style='text-align: center;'>We recommend you choosing <span style='color: red; font-weight: bold;'>{y_pred_6}</span> size!</h3>"
+        styled_text = f"<h3 style='text-align: center;'>You belong to <span style='color: red; font-weight: bold;'>{y_pred_6}</span> !</h3>"
         st.markdown(styled_text, unsafe_allow_html=True)
     
     elif num_of_features == '8 measurements':
@@ -179,12 +180,14 @@ if predict:
         X_8 = scaler_8.transform(X_8)
         y_8 = clf_8.predict(X_8)
         y_pred_8 = convert_prediction(np.argmax(y_8, axis=1))
-        styled_text = f"<h3 style='text-align: center;'>We recommend you choosing <span style='color: red; font-weight: bold;'>{y_pred_8}</span> size!</h3>"
+        # styled_text = f"<h3 style='text-align: center;'>We recommend you choosing <span style='color: red; font-weight: bold;'>{y_pred_8}</span> size!</h3>"
+        styled_text = f"<h3 style='text-align: center;'>You belong to <span style='color: red; font-weight: bold;'>{y_pred_8}</span> !</h3>"
         st.markdown(styled_text, unsafe_allow_html=True)
     else:
         X_21 = np.array([[h, w, bmi, vtn, vn, vcn, cn, cnnp, cnnt, cntp, cntt, ccnp, ccnt, snt, sndp, sndt, xup, xut, cl, ttp, ttt]])
         X_21 = scaler_21.transform(X_21)
         y_21 = clf_21.predict(X_21)
         y_pred_21 = convert_prediction(np.argmax(y_21, axis=1))
-        styled_text = f"<h3 style='text-align: center;'>We recommend you choosing <span style='color: red; font-weight: bold;'>{y_pred_21}</span> size!</h3>"
+        # styled_text = f"<h3 style='text-align: center;'>We recommend you choosing <span style='color: red; font-weight: bold;'>{y_pred_21}</span> size!</h3>"
+        styled_text = f"<h3 style='text-align: center;'>You belong to <span style='color: red; font-weight: bold;'>{y_pred_21}</span> !</h3>"
         st.markdown(styled_text, unsafe_allow_html=True)