Update app.py

app.py

@@ -42,13 +42,13 @@ def main():
     if data is not None:
         centroids, labels = apply_kmeans(data.values, k_value)
         
-        plt.figure(figsize=(8, 8))
-        plt.scatter(data['x'], data['y'], c=labels, cmap='viridis')
-        plt.scatter(centroids[:, 0], centroids[:, 1], s=200, c='red', marker='X')
-        plt.xlim(0, 100)
-        plt.ylim(0, 100)
-        plt.title(f"K-means clustering result for {dataset} Dataset (k={k_value})")
-        st.pyplot()
+        fig, ax = plt.subplots(figsize=(8, 8))
+        ax.scatter(data['x'], data['y'], c=labels, cmap='viridis')
+        ax.scatter(centroids[:, 0], centroids[:, 1], s=200, c='red', marker='X')
+        ax.set_xlim(0, 100)
+        ax.set_ylim(0, 100)
+        ax.set_title(f"K-means clustering result for {dataset} Dataset (k={k_value})")
+        st.pyplot(fig)
 
 if __name__ == "__main__":
     main()