Update app.py

app.py

@@ -38,9 +38,9 @@ pred_mean = k_test_train@k_inv@y_train
 pred_var = k_test_test - k_test_train@k_inv@k_test_train.T
 pred_std2 = 2 * (pred_var.diagonal() ** 0.5)
 
-ax.plot(x_test, pred_mean, label='predictive mean')
+ax.plot(x_test, pred_mean, label='predictive mean ($\\mu$)')
 ax.fill_between(x_test.ravel(), pred_mean.ravel()-pred_std2.ravel(), 
-                pred_mean.ravel()+pred_std2.ravel(), alpha=0.5, label='95% confidence')
+                pred_mean.ravel()+pred_std2.ravel(), alpha=0.5, label='$$\\mu \\pm 2\\sigma$$')
 ax.set_xlabel('x')
 ax.set_ylabel('y')
 ax.legend()