Update app.py

app.py

@@ -435,7 +435,7 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# ========== Title ==========
+# Title 
 st.title("✨ TensorFlow Playground")
 
 # Sidebar
@@ -536,7 +536,7 @@ if st.sidebar.button("🚀 Train Model"):
     hist = model.fit(x_train, y_train, epochs=epochs, validation_data=(x_test, y_test),
                      batch_size=32, verbose=0, callbacks=callbacks)
 
-# Neural Network Diagram
+    # Neural Network Diagram
     
     st.subheader("Neural Network Architecture")
     def visualize_nn(layers):