Update app.py

app.py

@@ -20,9 +20,9 @@ model_dir_white = model_white.download()
 model_white = joblib.load(model_dir_white + "/wine_white_model.pkl")
 print("White Model downloaded")
 
-def wine(alcohol, chlorides, density, type, volatil_acidity):
+def wine(alcohol, chlorides, density, type, volatil_acidity,fixed_acidity, citric_acid,total_sulfur_dioxide):
     print("Calling function")
-    df = pd.DataFrame([[type, alcohol, chlorides, density, volatil_acidity]], 
+    df = pd.DataFrame([[type, alcohol, chlorides, density, volatil_acidity,fixed_acidity, citric_acid,total_sulfur_dioxide]], 
                       columns=['type','alcohol','chlorides','density','volatil_acidity','fixed_acidity','citric_acid','total_sulfur_dioxide'])
     print("Predicting")
     print(df)