Enric Perpinyà Pitarch commited on
Commit
509a137
1 Parent(s): a715e4b
Files changed (1) hide show
  1. app.py +5 -17
app.py CHANGED
@@ -19,29 +19,17 @@ print("Model Loaded...")
19
  def wine(type,
20
  fixed_acidity,
21
  volatile_acidity,
22
- citric_acid,
23
- residual_sugar,
24
- chlorides,
25
- free_sulfur_dioxide,
26
- total_sulfur_dioxide,
27
- density,
28
- ph,
29
  sulphates,
30
- alcohol):
 
31
  print("Lets taste wine?")
32
 
33
- df = pd.DataFrame([[type, fixed_acidity, volatile_acidity,
34
- citric_acid, residual_sugar,
35
- chlorides, free_sulfur_dioxide,
36
- total_sulfur_dioxide,
37
- density, ph, sulphates, alcohol]],
38
  columns = ['type', 'fixed_acidity', 'volatile_acidity',
39
- 'citric_acid', 'residual_sugar', 'chlorides',
40
- 'free_sulfur_dioxide', 'total_sulfur_dioxide', 'density',
41
- 'ph', 'sulphates', 'alcohol'])
42
 
43
  print("Predicting...")
44
-
45
  res = model.predict(df)
46
  print(res)
47
 
 
19
  def wine(type,
20
  fixed_acidity,
21
  volatile_acidity,
 
 
 
 
 
 
 
22
  sulphates,
23
+ alcohol,
24
+ density):
25
  print("Lets taste wine?")
26
 
27
+ df = pd.DataFrame([[type, fixed_acidity, volatile_acidity, sulphates, alcohol, density]],
 
 
 
 
28
  columns = ['type', 'fixed_acidity', 'volatile_acidity',
29
+ 'sulphates', 'alcohol', 'density'])
 
 
30
 
31
  print("Predicting...")
32
+ print(df.head())
33
  res = model.predict(df)
34
  print(res)
35