Update pages/4_Model_Creation_and_Evaluation.py

pages/4_Model_Creation_and_Evaluation.py

@@ -1,6 +1,8 @@
 import streamlit as st
 import pandas as pd
 import numpy as np
+from io import StringIO
+import sys
 
 # Page configuration
 st.set_page_config(page_title="Predictive Modelling", layout="wide")
@@ -14,11 +16,12 @@ st.markdown(
 )
 # Flowchart title
 st.markdown(
-    '<div class="title">Model Creation Flow</div>', 
+    '"""
+    <h1 style="text-align: center; color: white;">Model Creation Flow</h1>
+    """, 
     unsafe_allow_html=True
 )
 
-# Flowchart GIF with 90% width
 st.markdown(
     """
     <div style="text-align: center;">
@@ -49,6 +52,7 @@ if df is not None:
 
     st.subheader("Shape of the Dataset:")
     st.write(df.shape)
+    )
 
 
 # Custom title styling