Update app.py

app.py

@@ -651,9 +651,10 @@ elif st.session_state.current_page == "Final Model":
         "The goal was to identify a model where both scores are closely aligned, ensuring minimal overfitting or underfitting.</p>",
         unsafe_allow_html=True,
     )
-    scatter_all = Image.open("images/bagging_trails.png")  
-    st.image(scatter_all, caption="Scatter Plot of All Trials", use_column_width=True)
     
+    st.image("images/bagging_trails.png", 
+         caption="All Trails", 
+         use_container_width=True)
   
     st.markdown("<h3>Training vs Test Score (First 50 Trials)</h3>", unsafe_allow_html=True)
     st.markdown(
@@ -661,9 +662,14 @@ elif st.session_state.current_page == "Final Model":
         "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)
+    
     # Final Model Summary
     st.markdown("<h3>Selected Best-Fit Model</h3>", unsafe_allow_html=True)
     st.markdown(