Fix progress bar error

app.py

@@ -76,15 +76,15 @@ if cost_function == 'RMSE-Loss':
      grad_loss = jax.grad(loss)
     
      # Perform gradient descent
-     for i in range(NUM_ITER):
+     for i in range(1,NUM_ITER+1):
          # Update progress bar.
-         progress_bar.progress(i)
+         progress_bar.progress(i+1)
 
          # Update parameters.
          w -= learning_rate * grad_loss(w)
 
          # Update status text.
-         if (i+1)%100==0:
+         if (i)%100==0:
              # report the loss at the current epoch
              status_text.text(
                  'Trained loss at epoch %s is %s' % (i, loss(w)))