Update app.py

app.py

@@ -630,22 +630,21 @@ elif st.session_state.current_page == "Final Model":
 
     # Title
     st.markdown("<h1 class='title'>Final Model</h1>", unsafe_allow_html=True)
-    
-    # Subtitle
-    st.markdown("<p class='subtitle'>Selecting the Best-Fit Model from Optuna Trials</p>", unsafe_allow_html=True)
 
-    # Explanation Box
+    st.markdown("<hr style='border:1px solid #ddd;'>", unsafe_allow_html=True)
+
     st.markdown(
         "<div class='box'>"
-        "<p>After experimenting with multiple trials using Optuna, we selected the best-fit model "
+        "<p><strong>After experimenting with multiple trials using Optuna, we selected the best-fit model "
         "by analyzing the training and test scores of different trials. "
-        "The following scatter plots provide insights into this selection process.</p>"
+        "The following scatter plots provide insights into this selection process.</strong></p>"
         "</div>",
         unsafe_allow_html=True,
     )
-    
-    
-    st.markdown("<h3>Training vs Test Score (All Trials)</h3>", unsafe_allow_html=True)
+
+    st.markdown("<hr style='border:1px solid #ddd;'>", unsafe_allow_html=True)
+
+    st.markdown("<h3 style='text-align: center;'>Training vs Test Score (All Trials)</h3>", unsafe_allow_html=True)
     st.markdown(
         "<p class='subtitle'>This scatter plot visualizes the training and test scores of all trials. "
         "The goal was to identify a model where both scores are closely aligned, ensuring minimal overfitting or underfitting.</p>",
@@ -655,23 +654,24 @@ elif st.session_state.current_page == "Final Model":
     st.image("images/bagging_trails.png", 
          caption="All Trails", 
          use_container_width=True)
+
+    st.markdown("<hr style='border:1px solid #ddd;'>", unsafe_allow_html=True)
   
-    st.markdown("<h3>Training vs Test Score (First 50 Trials)</h3>", unsafe_allow_html=True)
+    st.markdown("<h3 style='text-align: center;'>Training vs Test Score (First 50 Trials)</h3>", unsafe_allow_html=True)
     st.markdown(
         "<p class='subtitle'>By filtering the first 50 trials, we focused on models that demonstrated balanced performance. "
         "The best-fit model was selected by ensuring that the training and test scores are close to each other.</p>",
         unsafe_allow_html=True,
     )
-    
-    scatter_50 = Image.open("images/bagging_50trails.png")  
-    st.image(scatter_50, caption="Scatter Plot of First 50 Trials", use_column_width=True)
 
     st.image("images/bagging_50trails.png", 
      caption="50 Trails", 
      use_container_width=True)
+
+    st.markdown("<hr style='border:1px solid #ddd;'>", unsafe_allow_html=True)
     
-    # Final Model Summary
-    st.markdown("<h3>Selected Best-Fit Model</h3>", unsafe_allow_html=True)
+    st.markdown("<h3 style='text-align: center;'>Selected Best-Fit Model</h3>", unsafe_allow_html=True)
+
     st.markdown(
         "<div class='box'>"
         "<ul>"