edit code snippet

app.py

@@ -62,8 +62,11 @@ else:
 st.markdown('The training loop:')
 
 code = '''NUM_ITER = 1000
-          for i in range(NUM_ITER):
-               w -= learning_rte * grad_loss(w)'''
+     # initialize parameters
+     w = np.array([3., -2., -8.])
+     for i in range(NUM_ITER):
+          # update parameter 
+          w -= learning_rte * grad_loss(w)'''
      
 st.code(code, language='python')