Enric Perpinyà Pitarch commited on
Commit
e7a3d57
·
1 Parent(s): 32f504b

First commit

Browse files
Files changed (2) hide show
  1. app.py +72 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from PIL import Image
3
+ import requests
4
+ import hopsworks as hw
5
+ import joblib
6
+ import pandas as pd
7
+
8
+ project = hw.login()
9
+ fs = project.get_feature_store()
10
+
11
+ mr = project.get_model_registry()
12
+ model = mr.get_model("wine_model", version=1)
13
+ model_dir = model.download()
14
+
15
+ model = joblib.load(model_dir+'/wine_model.pkl')
16
+ print("Model Loaded...")
17
+
18
+ def wine(type,
19
+ fixed_acidity,
20
+ volatile_acidity,
21
+ citric_acid,
22
+ residual_sugar,
23
+ chlorides,
24
+ free_sulfur_dioxide,
25
+ total_sulfur_dioxide,
26
+ density,
27
+ ph,
28
+ sulphates,
29
+ alcohol):
30
+ print("Lets taste wine?")
31
+
32
+ df = pd.DataFrame([[type, fixed_acidity, volatile_acidity,
33
+ citric_acid, residual_sugar,
34
+ chlorides, free_sulfur_dioxide,
35
+ total_sulfur_dioxide,
36
+ density, ph, sulphates, alcohol]],
37
+ columns = ['type', 'fixed_acidity', 'volatile_acidity',
38
+ 'citric_acid', 'residual_sugar', 'chlroides',
39
+ 'free_sulfur_dioxide', 'total_sulfur_dioxide',
40
+ 'ph', 'sulphates', 'alcohol'])
41
+
42
+ print("Predicting...")
43
+ print(df)
44
+
45
+ res = model.predict(df)
46
+ print(res)
47
+
48
+ return int(res.round()) + 3
49
+
50
+ demo = gr.Interface(
51
+ fn = wine,
52
+ title = 'Wine Quality prediction',
53
+ description = '',
54
+ allow_flagging = 'never',
55
+ inputs = [
56
+ gr.inputs.Number(default=0, label="type"),
57
+ gr.inputs.Number(default=0.67, label="fixed_acidity"),
58
+ gr.inputs.Number(default=-0.38, label="volatile_acidity"),
59
+ gr.inputs.Number(defalt=0.55, label="citric_acid"),
60
+ gr.inputs.Number(default=0.4, label="residual_sugar"),
61
+ gr.inputs.Number(default=-0.18, label="chlroides"),
62
+ gr.inputs.Number(default=-0.0033, label="free_sulfur_dioxide"),
63
+ gr.inputs.Number(default=-0.3, label="total_sulfur_dioxide"),
64
+ gr.inputs.Number(default=0.19, label="density"),
65
+ gr.inputs.Number(default=0.22, label="ph"),
66
+ gr.inputs.Number(default=-0.62, label="sulphates"),
67
+ gr.inputs.Number(default=-0.37, label="alcohol")
68
+ ],
69
+ outputs = ['Quality']# output's an integer from 3-9
70
+ )
71
+
72
+ demo.launch(debug=True)
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ hopsworks
2
+ joblib
3
+ scikit-learn==1.1.1