Fix indentation error

app.py

@@ -71,29 +71,29 @@ if cost_function == 'RMSE-Loss':
      st.latex(r'''\ell(X, y, w)= \frac{1}{m}\sum_1^m (\hat{y}_i - y_i)^2''')
      
      
-    progress_bar = st.progress(0)
-    status_text = st.empty()
-    grad_loss = jax.grad(loss)
+     progress_bar = st.progress(0)
+     status_text = st.empty()
+     grad_loss = jax.grad(loss)
     
-    # Perform gradient descent
-    for i in range(NUM_ITER):
-        # Update progress bar.
-        progress_bar.progress(i + 1)
-
-        # Update parameters.
-        w -= learning_rate * grad_loss(w)
-
-        # Update status text.
-        if (i+1)%100==0:
-            status_text.text(
-                'Trained loss is: %s' % loss(w))
-
-            # add a line to the plot
-            plt.plot(X[jnp.dot(X, w).argsort(), 1], jnp.dot(X, w).sort(), '--')
-      print(f"Trained loss at epoch #{i}:", loss(w))
-    plt.plot(X[jnp.dot(X, w).argsort(), 1], jnp.dot(X, w).sort(), '--', label='Final line')
-
-    status_text.text('Done!')
+     # Perform gradient descent
+     for i in range(NUM_ITER):
+         # Update progress bar.
+         progress_bar.progress(i + 1)
+
+         # Update parameters.
+         w -= learning_rate * grad_loss(w)
+
+         # Update status text.
+         if (i+1)%100==0:
+             # report the loss at the current epoch
+             status_text.text(
+                 'Trained loss is: %s' % loss(w))
+
+             # add a line to the plot
+             plt.plot(X[jnp.dot(X, w).argsort(), 1], jnp.dot(X, w).sort(), 'c--')
+     # Plot the final line
+     plt.plot(X[jnp.dot(X, w).argsort(), 1], jnp.dot(X, w).sort(), 'k-', label='Final line'
+     status_text.text('Done!')
 
 
 else: