Upload app.py

app.py

@@ -25,17 +25,20 @@ def main():
         clicked = st.form_submit_button("Predict quality")
         
         if clicked:
-            result=model.predict(pd.DataFrame({"fixed_acidity": [fixed_acidity],
-                                               "volatile_acidity": [volatile_acidity],
-                                               "citric_acid": [citric_acid],
-                                               "residual_sugar": [residual_sugar],
-                                               "chlorides": [chlorides],
-                                               "free_sulfur_dioxide": [free_sulfur_dioxide],
-                                               "total_sulfur_dioxide": [total_sulfur_dioxide],
-                                               "density": [density],
-                                               "ph": [ph],
-                                               "sulphates": [sulphates],
-                                               "alcohol": [alcohol]}))
+            result = model.predict(pd.DataFrame({
+                "fixed acidity": [fixed_acidity],
+                "volatile acidity": [volatile_acidity],
+                "citric acid": [citric_acid],
+                "residual sugar": [residual_sugar],
+                "chlorides": [chlorides],
+                "free sulfur dioxide": [free_sulfur_dioxide],
+                "total sulfur dioxide": [total_sulfur_dioxide],
+                "density": [density],
+                "pH": [ph],
+                "sulphates": [sulphates],
+                "alcohol": [alcohol]
+            }))
+
             predicted_quality_rank = result[0]
             st.success('The predicted wine quality ranking is {}'.format(predicted_quality_rank))