Update app.py

app.py

@@ -72,7 +72,6 @@ model = load_model("final_model_kfold.h5", custom_objects={'f1_weighted': f1_wei
 
 from sklearn.preprocessing import LabelEncoder
 encoder =  LabelEncoder()
-print(test_data.columns)
 y_test = encoder.fit_transform(test_data["prdtypecode"])
 class_labels = encoder.classes_
 label_size = 27
@@ -80,7 +79,15 @@ label_size = 27
 
 
 ####### Page principale
-st.set_page_config(layout="wide")
+st.set_page_config(layout="wide",page_title="Rakuten Challenge")
+hide_default_format = """
+       <style>
+       #MainMenu {visibility: hidden; }
+       footer {visibility: hidden;}
+       </style>
+       """
+st.markdown(hide_default_format, unsafe_allow_html=True)
+
 st.markdown('<style>' + open('./style.css').read() + '</style>', unsafe_allow_html=True)
 
 st.title("Rakuten Challenge")
@@ -156,7 +163,7 @@ elif tabs == "Preprocessing":
     detection_langage_et_traduction(st, extract_data, sum_data)
 
 elif tabs == "Modèle":
-    presentation_modele(st, test_data, model,class_labels,y_test)
+    presentation_modele(st, test_data, model,class_labels,y_test,encoder)
 
 elif tabs == "Pistes exploratoires":
     st.write("# Pistes exploratoires")