Update app.py

app.py

@@ -131,7 +131,7 @@ if st.sidebar.button('Submit'):
 
     # Data visualization
     st.subheader("Visualization of Data Points with Class Labels")
-    fig, ax = plt.subplots(figsize=(10, 4))
+    fig, ax = plt.subplots(figsize=(10, 6))
     if problem_type in ['Classification', 'Moons', 'Circles']:
         sns.scatterplot(x=X[:, 0], y=X[:, 1], hue=y, ax=ax)
     else: