Update Home.py

Home.py

@@ -169,14 +169,24 @@ if predict:
         X_6 = np.array([[vtn, vn, vcn, cl, ttp, ttt]])
         X_6 = scaler_6.transform(X_6)
         y_6 = clf_6.predict(X_6)
-        st.success(f'We recommend you choosing **{convert_prediction(y_6)}** size!')
+        y_pred = convert_prediction(y_6)
+        # st.success(f'We recommend you choosing **{convert_prediction(y_6)}** size!')
+        styled_text = f"<h3 style='text-align: center;'>We recommend you choosing <span style='color: red; font-weight: bold;'>{y_pred}</span> size!</h3>"
+        st.markdown(styled_text, unsafe_allow_html=True)
+    
     elif num_of_features == '8 measurements':
         X_8 = np.array([[ttp, cl, cnnp, vn, vtn, cnnt, cntp, vcn]])
         X_8 = scaler_8.transform(X_8)
         y_8 = clf_8.predict(X_8)
-        st.success(f'We recommend you choosing **{convert_prediction(np.argmax(y_8, axis=1))}** size!')
+        # st.success(f'We recommend you choosing **{convert_prediction(np.argmax(y_8, axis=1))}** size!')
+        y_pred = convert_prediction(y_8)
+        # st.success(f'We recommend you choosing **{convert_prediction(y_6)}** size!')
+        styled_text = f"<h3 style='text-align: center;'>We recommend you choosing <span style='color: red; font-weight: bold;'>{y_pred}</span> size!</h3>"
     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)
-        st.success(f'We recommend you choosing **{convert_prediction(np.argmax(y_21, axis=1))}** size!')
+        # st.success(f'We recommend you choosing **{convert_prediction(np.argmax(y_21, axis=1))}** size!')
+        y_pred = convert_prediction(y_21)
+        # st.success(f'We recommend you choosing **{convert_prediction(y_6)}** size!')
+        styled_text = f"<h3 style='text-align: center;'>We recommend you choosing <span style='color: red; font-weight: bold;'>{y_pred}</span> size!</h3>"