Update app.py

app.py

@@ -81,11 +81,16 @@ class KMeansTransformer(BaseEstimator, TransformerMixin):
 # Loading the model
 pipe = joblib.load('wine_quality_prediction.pkl')
 
-# Define a function to take user inputs
+input_features = [
+    "fixed_acidity", "volatile_acidity", "citric_acid", "residual_sugar", 
+    "chlorides", "free_sulfur_dioxide", "total_sulfur_dioxide", "density", 
+    "pH", "sulphates", "alcohol"
+]
+
 def get_user_input():
     input_dict = {}
-    for feat in selected_features:
-        input_value = st.number_input(f"Enter {feat} value", value=0.0, step=0.01)
+    for feat in input_features:
+        input_value = st.number_input(f"Enter {feat}", value=0.0, step=0.01)
         input_dict[feat] = input_value
     return pd.DataFrame([input_dict])