Update app.py

app.py

@@ -21,15 +21,14 @@ model_dir_white = model_white.download()
 model_white = joblib.load(model_dir_white + "/wine_white_model.pkl")
 print("White Model downloaded")
 
-def wine(quality,fixed_acidity, volatil_acidity, citric_acid, chlorides,
-                        total_sulfur_dioxide, density, alcohol):
+def wine(category, alcohol, chlorides, density, volatil_acidity,fixed_acidity, citric_acid,total_sulfur_dioxide):
     print("Calling function")
-    df = pd.DataFrame([[type,fixed_acidity, volatil_acidity, citric_acid, chlorides,total_sulfur_dioxide, density, alcohol]], 
-                      columns=['type','fixed_acidity', 'volatil_acidity', 'citric_acid', 'chlorides','total_sulfur_dioxide', 'density', 'alcohol'])
+    df = pd.DataFrame([[alcohol, chlorides, density, volatil_acidity,fixed_acidity, citric_acid,total_sulfur_dioxide]], 
+                      columns=['alcohol','chlorides','density','volatil_acidity','fixed_acidity','citric_acid','total_sulfur_dioxide'])
     print("Predicting")
     print(df)
     
-    if type == "red":
+    if category == "red":
          res = model_red.predict(df) 
          print(res)
          wine_url = "https://raw.githubusercontent.com/Anniyuku/wine_quality/main/" + res[0] + ".png"
@@ -44,17 +43,17 @@ def wine(quality,fixed_acidity, volatil_acidity, citric_acid, chlorides,
 demo = gr.Interface(
     fn=wine,
     title="Wine Predictive Analytics",
-    description="Experiment with fixed_acidity, volatil_acidity, citric_acid, chlorides,total_sulfur_dioxide, density, alcohol to predict which wine quality it is.",
+    description="Experiment with type, alcohol, chlorides, density, volatil_acidity, fixed_acidity, citric_acid, total_sulfur_dioxide to predict which flower it is.",
     allow_flagging="never",
     inputs=[
-        gr.inputs.Radio(choices=['red', 'white'], label='type'),
-        gr.inputs.Number(default=9.00, label="alcohol"),
-        gr.inputs.Number(default=0.60, label="chlorides"),
-        gr.inputs.Number(default=1.00, label="density"),
-        gr.inputs.Number(default=1.00, label="volatil_acidity"),
-        gr.inputs.Number(default=1.00, label="fixed_acidity"),
-        gr.inputs.Number(default=1.00, label="citric_acid"),
-        gr.inputs.Number(default=1.00, label="total_sulfur_dioxide"),
+        gr.inputs.Radio(choices=["white","red"], label='category'),
+        gr.inputs.Number(default=12.4, label="alcohol"),
+        gr.inputs.Number(default=0.04, label="chlorides"),
+        gr.inputs.Number(default=0.99, label="density"),
+        gr.inputs.Number(default=0.16, label="volatil_acidity"),
+        gr.inputs.Number(default=6.60, label="fixed_acidity"),
+        gr.inputs.Number(default=0.40, label="citric_acid"),
+        gr.inputs.Number(default=143, label="total_sulfur_dioxide"),
         ],
     outputs=gr.Image(type="pil"))