Fix code

app.py

@@ -6,12 +6,12 @@ number_of_observations = st.slider('Number of observations', min_value=50, max_v
 X = np.column_stack((np.ones(number_of_observations), 
                      np.random.random(number_of_observations))     
                                     
-fig = plt.figure(dpi=80, figsize=(7,7))
-ax = fig.add_subplot(111)
+
+fig, ax = plt.subplots()
 ax.set_xlim((0,1))
 ax.set_ylim((-5,20))
-plt.scatter(X[:,1], y, c='r', edgecolors='#fda172')
-line_thickness = 2
-line, = ax.plot([], [], lw=line_thickness)
+ax.scatter(X[:,1], y, c='r', edgecolors='#fda172')
+#line_thickness = 2
+#line, = ax.plot([], [], lw=line_thickness)
 st.pyplot(fig)
 st.write(X[:5, :])