Update app.py
Browse files
app.py
CHANGED
@@ -15,8 +15,8 @@ model_dir = model.download()
|
|
15 |
model = joblib.load(model_dir + "/wine_model.pkl")
|
16 |
print("Model downloaded")
|
17 |
|
18 |
-
def wine(Type,
|
19 |
-
total_sulfur_dioxide, density, ph,
|
20 |
#Maps type to int with default white
|
21 |
if Type=="red":
|
22 |
Type=1
|
@@ -53,7 +53,6 @@ demo = gr.Interface(
|
|
53 |
gr.inputs.Number(default=3.2, label="pH"),
|
54 |
gr.inputs.Number(default=0.53, label="Sulphates"),
|
55 |
gr.inputs.Number(default=10.5, label="Alcohol precentage"),
|
56 |
-
gr.inputs.Number(default=0.056, label="Chlorides"),
|
57 |
],
|
58 |
outputs=gr.Label())
|
59 |
|
|
|
15 |
model = joblib.load(model_dir + "/wine_model.pkl")
|
16 |
print("Model downloaded")
|
17 |
|
18 |
+
def wine(Type, volatile_acidity, citric_acid, residual_sugar, chlorides, free_sulfur_dioxide,
|
19 |
+
total_sulfur_dioxide, density, ph, sulphates, alcohol):
|
20 |
#Maps type to int with default white
|
21 |
if Type=="red":
|
22 |
Type=1
|
|
|
53 |
gr.inputs.Number(default=3.2, label="pH"),
|
54 |
gr.inputs.Number(default=0.53, label="Sulphates"),
|
55 |
gr.inputs.Number(default=10.5, label="Alcohol precentage"),
|
|
|
56 |
],
|
57 |
outputs=gr.Label())
|
58 |
|