Fix progress index

app.py

@@ -78,7 +78,7 @@ if cost_function == 'RMSE-Loss':
      # Perform gradient descent
      for i in range(NUM_ITER):
          # Update progress bar.
-         progress_bar.progress(i + 1)
+         progress_bar.progress(i)
 
          # Update parameters.
          w -= learning_rate * grad_loss(w)
@@ -87,7 +87,7 @@ if cost_function == 'RMSE-Loss':
          if (i+1)%100==0:
              # report the loss at the current epoch
              status_text.text(
-                 'Trained loss is: %s' % loss(w))
+                 'Trained loss at epoch %s is %s' % (i, loss(w)))
 
              # add a line to the plot
              plt.plot(X[jnp.dot(X, w).argsort(), 1], jnp.dot(X, w).sort(), 'c--')