Update app.py

app.py

@@ -39,8 +39,8 @@ def run_linear_regression(X, y, alpha_prior, beta_prior, sigma_prior):
         alpha_i = samples['alpha'][i]
         beta_i = samples['beta'][i]
         ax.plot(X, alpha_i + beta_i * X, color=light_color)
-    alpha_i = samples['alpha'][499]
-    beta_i = samples['beta'][499]
+    alpha_i = samples['alpha'][498]
+    beta_i = samples['beta'][498]
     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')