Update app.py

app.py

@@ -70,7 +70,6 @@ if st.button("Set Up") or 'func' not in st.session_state or 'points' not in st.s
     except Exception as e:
         st.error(f"Error setting up function: {e}")
 
-# ------------------------ ITERATION STEP ------------------------
 def gradient_step(x_val, func, lr):
     x = torch.tensor([x_val], dtype=torch.float32, requires_grad=True)
     y = func(x)
@@ -101,7 +100,7 @@ if 'func' in st.session_state:
         except Exception as e:
             st.error(f"Error in multiple iterations: {e}")
 
-# ------------------------ PLOTTING ------------------------
+
 if 'func' in st.session_state and len(st.session_state.points) > 0:
     try:
         x_val = np.linspace(-10, 10, 500)