Update app.py

app.py

@@ -15,15 +15,9 @@ from keras.losses import MeanSquaredError, BinaryCrossentropy
 from keras.regularizers import l2, l1
 from keras.callbacks import Callback
 
-# 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")
+# Debug versions
+st.write(f"TensorFlow version: {tf.__version__}")
+st.write(f"Keras version: {tf.keras.__version__}")
 
 # Set wide layout and TensorFlow Playground CSS
 st.set_page_config(layout="wide")