datnguyentien204
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import warnings
|
|
6 |
|
7 |
warnings.filterwarnings("ignore", category=UserWarning)
|
8 |
|
9 |
-
st.set_page_config(page_title="ViWoBra - Hệ thống phân loại
|
10 |
|
11 |
def load_essential_models(scaler_6_path, scaler_8_path, scaler_full_path, clf_6_path, clf_8_path, clf_full_path):
|
12 |
scaler_6 = pickle.load(open(file=scaler_6_path, mode='rb'))
|
@@ -25,7 +25,7 @@ def convert_prediction(prediction):
|
|
25 |
else:
|
26 |
return 'the large breast group'
|
27 |
|
28 |
-
st.markdown("<h1 style='text-align: center;'>Vietnamese Woman
|
29 |
|
30 |
scaler_6, scaler_8, scaler_21, clf_6, clf_8, clf_21 = load_essential_models(scaler_6_path='./Models/6/scaler.pkl', scaler_8_path='./Models/8/scaler.pkl', scaler_full_path='./Models/21/scaler.pkl', clf_6_path='./Models/6/svc_fs_tune.pkl', clf_8_path='./Models/8/ANN_8.h5', clf_full_path='./Models/21/ANNs_full.h5')
|
31 |
|
|
|
6 |
|
7 |
warnings.filterwarnings("ignore", category=UserWarning)
|
8 |
|
9 |
+
st.set_page_config(page_title="ViWoBra - Hệ thống phân loại ngực phụ nữ Việt Nam", page_icon="💮", layout='wide', initial_sidebar_state="collapsed")
|
10 |
|
11 |
def load_essential_models(scaler_6_path, scaler_8_path, scaler_full_path, clf_6_path, clf_8_path, clf_full_path):
|
12 |
scaler_6 = pickle.load(open(file=scaler_6_path, mode='rb'))
|
|
|
25 |
else:
|
26 |
return 'the large breast group'
|
27 |
|
28 |
+
st.markdown("<h1 style='text-align: center;'>Vietnamese Woman Breast Classifier</h1>", unsafe_allow_html=True)
|
29 |
|
30 |
scaler_6, scaler_8, scaler_21, clf_6, clf_8, clf_21 = load_essential_models(scaler_6_path='./Models/6/scaler.pkl', scaler_8_path='./Models/8/scaler.pkl', scaler_full_path='./Models/21/scaler.pkl', clf_6_path='./Models/6/svc_fs_tune.pkl', clf_8_path='./Models/8/ANN_8.h5', clf_full_path='./Models/21/ANNs_full.h5')
|
31 |
|