Update app.py

app.py

@@ -79,7 +79,7 @@ if uploaded_file is not None:
         y = df_cleaned[target]
         
         # Determine if the target is continuous or categorical
-        is_classification = y.nunique() <= 10  # If target has fewer than or equal to 10 unique values, treat as classification
+        is_classification = len(y.unique()) <= 10  # If target has fewer than or equal to 10 unique values, treat as classification
         
         # Ensure there is enough data before proceeding with train-test split
         if len(X) == 0 or len(y) == 0:
@@ -165,18 +165,15 @@ if uploaded_file is not None:
                     mime="image/png"
                 )
 
-            # Display Histograms of Numerical Columns
-            st.subheader("Histograms of Numerical Columns")
-            for column in df_cleaned.select_dtypes(include=['number']).columns:
-                fig, ax = plt.subplots(figsize=(5, 4))  # Small graph
-                df_cleaned[column].plot(kind="hist", bins=20, ax=ax, title=column)
-                st.pyplot(fig)
-                # Save and provide download option for histogram
-                fig.savefig(f"/tmp/{column}_histogram.png")
-                with open(f"/tmp/{column}_histogram.png", "rb") as f:
-                    st.download_button(
-                        label=f"Download {column} Histogram",
-                        data=f,
-                        file_name=f"{column}_histogram.png",
-                        mime="image/png"
-                    )
+            # Pair plot of numerical columns to visualize relationships
+            st.subheader("Pair Plot of Numerical Columns")
+            pair_plot = sns.pairplot(df_cleaned[features])  # Generate pair plot for features
+            st.pyplot(pair_plot)
+            pair_plot.savefig("/tmp/pair_plot.png")
+            with open("/tmp/pair_plot.png", "rb") as f:
+                st.download_button(
+                    label="Download Pair Plot",
+                    data=f,
+                    file_name="pair_plot.png",
+                    mime="image/png"
+                )