Update app.py

app.py

@@ -35,10 +35,13 @@ def run_linear_regression(X, y, alpha_prior, beta_prior, sigma_prior):
     fig, ax = plt.subplots(figsize=(8, 6))
     ax.scatter(X, y, color='blue', alpha=0.5, label='data')
     light_color = (1.0, 0.5, 0.5, 0.7)
-    for i in range(500):
+    for i in range(499):
         alpha_i = samples['alpha'][i]
         beta_i = samples['beta'][i]
-        ax.plot(X, alpha_i + beta_i * X, color=light_color,label='MCMC samples')
+        ax.plot(X, alpha_i + beta_i * X, color=light_color)
+    alpha_i = samples['alpha'][499]
+    beta_i = samples['beta'][499]
+    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)