Update app.py

app.py

@@ -15,7 +15,7 @@ categorical_features = joblib.load("categorical_features.joblib")
 encoder = joblib.load("encoder.joblib")
 
 
-option = st.sidebar.selectbox("Which dashboard?", ("Model information", "Stroke prediction"))
+option = st.sidebar.selectbox("Which dashboard?", ("Information about training data","Model information", "Stroke prediction"))
 st.title(option)
 
 def get_pred():
@@ -117,4 +117,7 @@ if option == "Model information":
 
 
     st.subheader("Shap values visualization of how features contribute to model prediction")
-    st.image(r'lgbm_model_shap_evaluation.png')
+    st.image(r'lgbm_model_shap_evaluation.png')
+    st.subheader("Confusion matrix of predictions made by a Light GradientBoost Classifier and it's ROC_Curve")
+    
+    st.image([r'lgbm_model_shap_evaluation.png',r'lgbm_model_shap_evaluation.png])