Update app.py

app.py

@@ -38,7 +38,7 @@ def run_linear_regression(X, y, alpha_prior, beta_prior, sigma_prior):
     for i in range(500):
         alpha_i = samples['alpha'][i]
         beta_i = samples['beta'][i]
-        ax.plot(X, alpha_i + beta_i * X, color=light_color)
+        ax.plot(X, alpha_i + beta_i * X, color=light_color,label='MCMC samples')
     ax.plot(X, np.mean(samples['alpha']) + np.mean(samples['beta']) * X, color='red', label='mean')
     ax.legend(loc='upper left')
     st.pyplot(fig)