Update app.py

app.py

@@ -15,9 +15,15 @@ from keras.losses import MeanSquaredError, BinaryCrossentropy
 from keras.regularizers import l2, l1
 from keras.callbacks import Callback
 
-# Debug versions
-st.write(f"TensorFlow version: {tf.__version__}")
-st.write(f"Keras version: {tf.keras.__version__}")
+# Check TensorFlow and Keras versions with fallback
+try:
+    tf_version = tf.__version__
+    keras_version = tf.keras.__version__ if hasattr(tf.keras, '__version__') else "Keras version not available (bundled with TensorFlow)"
+    st.write(f"TensorFlow version: {tf_version}")
+    st.write(f"Keras version: {keras_version}")
+except AttributeError as e:
+    st.error(f"Error checking versions: {e}")
+    st.write("Falling back to default versions: TensorFlow ~2.10, Keras ~2.10")
 
 # Set wide layout and TensorFlow Playground CSS
 st.set_page_config(layout="wide")