Update src/streamlit_app.py

src/streamlit_app.py

@@ -25,15 +25,14 @@ def get_data():
 df = get_data()
 st.write("Dataset loaded! Here's a preview:")
 st.dataframe(df.head())
-
 # -------------------------- Preprocessing --------------------------
-# Drop ID (not useful) and separate features/target
-X = df.drop(["quality", "Id"], axis=1)
+X = df.drop("quality", axis=1)       # ← fixed: no "Id" column exists
 y = df["quality"]
 
-# Make it a binary classification (good ≥ 6, bad < 6) - common practice
+# Make it binary classification: good (≥6) vs bad (<6)
 y = (y >= 6).astype(int)
 
+
 # Train-test split
 X_train, X_test, y_train, y_test = train_test_split(
     X, y, test_size=0.2, random_state=42, stratify=y