Update app.py

app.py

@@ -15,7 +15,10 @@ from keras.losses import MeanSquaredError, BinaryCrossentropy
 from keras.regularizers import l2, l1
 from keras.callbacks import Callback
 
-# Check TensorFlow and Keras versions with fallback
+# Set page config as the first Streamlit command
+st.set_page_config(layout="wide")
+
+# Check TensorFlow and Keras versions with fallback (after set_page_config)
 try:
     tf_version = tf.__version__
     # Try multiple ways to get Keras version, accounting for TensorFlow integration
@@ -32,8 +35,7 @@ except AttributeError as e:
     st.error(f"Error checking versions: {e}")
     st.write("Falling back to default versions: TensorFlow ~2.15, Keras ~2.15")
 
-# Set wide layout and TensorFlow Playground CSS
-st.set_page_config(layout="wide")
+# Set TensorFlow Playground CSS
 st.markdown("""
     <style>
     .stApp {