Update app.py

app.py

@@ -6,6 +6,8 @@ from sklearn.model_selection import train_test_split
 from sklearn.linear_model import LogisticRegression
 from sklearn.preprocessing import LabelEncoder, StandardScaler
 from sklearn.metrics import classification_report, confusion_matrix, roc_curve, auc
+from io import StringIO
+
 
 # Load the dataset
 @st.cache
@@ -25,9 +27,10 @@ if choice == "Data Overview":
     st.write(data.head())
 
     st.write("### Dataset Information")
-    buffer = []
+    # Create a buffer for the info output
+    buffer = StringIO()
     data.info(buf=buffer)
-    s = "\n".join(buffer)
+    s = buffer.getvalue()  # Get the output from the buffer
     st.text(s)
 
     st.write("### Dataset Statistics")