Update app.py

app.py

@@ -89,26 +89,26 @@ input_features = [
 
 def get_user_input():
     input_dict = {}
-    for feat in input_features:
-        input_value = st.number_input(f"Enter {feat}", value=0.0, step=0.01)
-        input_dict[feat] = input_value
 
-    user_input = pd.DataFrame([input_dict])
+    
+    with st.form(key='my_form'):
+        for feat in input_features:
+            input_value = st.number_input(f"Enter value for {feat}", value=0.0, step=0.01)
+            input_dict[feat] = input_value
 
-    # Debug: print out the user input
-    st.write(user_input)
+       
+        submit_button = st.form_submit_button(label='Submit')
 
-    return user_input
+    return pd.DataFrame([input_dict]), submit_button
 
-# Title
-st.title("Wine Quality Prediction")
 
-# Get user inputs
-user_input = get_user_input()
+user_input, submit_button = get_user_input()
 
-# Make predictions
-prediction = pipe.predict(user_input)
+# When the 'Submit' button is pressed, perform the prediction
+if submit_button:
+    # Predict wine quality
+    prediction = pipe.predict(user_input)
 
-# Display the result
-st.subheader("Prediction")
-st.write(prediction[0])
+    # Display the prediction
+    st.header("Predicted Quality")
+    st.write(prediction)