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 |
|