Update app.py

app.py

@@ -7,8 +7,8 @@ from matplotlib import rc
 st.subheader("Bayesian Coin Toss")
 st_col = st.columns(1)[0]
 
-n_samples = st.slider('N_samples', 2, 40, 1)
-n_heads = st.slider('N_heads', 2, 40, 1)
+N = st.slider('N_samples', 2, 40, 1)
+h = st.slider('N_heads', 2, 40, 1)
 alpha = st.slider('Alpha', 0.5, 5.0, 0.5)
 beta = st.slider('Beta', 0.5, 5.0, 0.5)
 
@@ -22,8 +22,6 @@ def Bernoulli(theta, N, h):
     return (theta ** h) * ((1-theta) ** (N-h))
 
 Likelihood = [Bernoulli(t,N,h) for t in theta]
-
-ax.cla();axs.cla()
 ax.plot(theta, Likelihood, label='Likelihood',color='b');
 axs.plot(theta, scipy.stats.beta.pdf(theta, alpha,beta), label='Prior',color='k');
 ax.set_xlabel('p(head)');
@@ -44,6 +42,9 @@ axs.tick_params(axis='y', colors='r')
 ax.set_ylabel('Likelihood',color='b')
 axs.set_ylabel('Prior/Posterior', color='r', rotation=270, labelpad=30)
 
+ax.spines['top'].set_visible(False)
+axs.spines['top'].set_visible(False)
+
 with st_col:
     st.pyplot(fig)