Added a bit of padding to prevent label from overlapping graph title.

app.py

@@ -15,12 +15,13 @@ def initial_points(X, y_true, n_components, n_clust):
     centers_init, indices = kmeans_plusplus(X, n_clusters=n_clust, random_state=0)
     
     # Plot init seeds along side sample data
-    init_points_plot = plt.figure()
+    init_points_plot, ax = plt.subplots()
 
     for k in range(n_components):
         cluster_data = y_true == k
         plt.scatter(X[cluster_data, 0], X[cluster_data, 1], marker=".", s=10)
 
+    plt.subplots_adjust(top=0.8)
     plt.scatter(centers_init[:, 0], centers_init[:, 1], c="b", s=50)
     plt.title("K-Means++ Initialization")
     plt.xticks([])
@@ -31,11 +32,12 @@ def one_step(X, n_clust):
     kmeans = KMeans(n_clusters=n_clust, max_iter=1, n_init=1, random_state=0).fit(X)
     y_hat = kmeans.predict(X)
 
-    one_step = plt.figure()
+    one_step, ax = plt.subplots()
     plt.scatter(X[:, 0], X[:, 1], marker=".", s=10, c=y_hat)
     centers = kmeans.cluster_centers_
     plt.scatter(centers[:, 0], centers[:, 1], c="b", s=50)
 
+    plt.subplots_adjust(top=0.8)
     plt.title("K-Means After One Step")
     plt.xticks([])
     plt.yticks([])