bqpxben commited on
Commit
1debb2e
·
1 Parent(s): 51f4480

Upload 3 files

Browse files
Files changed (3) hide show
  1. app.py +43 -0
  2. model.joblib +3 -0
  3. requirements.txt +5 -0
app.py ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import joblib
2
+ import pandas as pd
3
+ import streamlit as st
4
+
5
+
6
+ model = joblib.load('model.joblib')
7
+
8
+ def main():
9
+ st.title("Wine Quality Analysis")
10
+
11
+ with st.form("questionaire"):
12
+
13
+ fixed_acidity = st.slider("Fixed Acidity", min_value=0.0, max_value=20.0, step=0.01)
14
+ volatile_acidity = st.slider("Volatile Acidity", min_value=0.0, max_value=20.0, step=0.01)
15
+ citric_acid = st.slider("Citric Acid", min_value=0.0, max_value=1.0, step=0.01)
16
+ residual_sugar = st.slider("Residual Sugar", min_value=0.0, max_value=20.0, step=0.01)
17
+ chlorides = st.slider("Chlorides", min_value=0.0, max_value=1.0, step=0.01)
18
+ free_sulfur_dioxide = st.slider("Free Sulfur Dioxide", min_value=0.0, max_value=100.0, step=0.01)
19
+ total_sulfur_dioxide = st.slider("Total Sulfur Dioxide", min_value=0.0, max_value=500.0, step=0.01)
20
+ density = st.slider("Density", min_value=0.0, max_value=10.0, step=0.01)
21
+ ph = st.slider("pH", min_value=1.0, max_value=14.0, step=0.01)
22
+ sulphates = st.slider("Sulphates", min_value=0.0, max_value=20.0, step=0.01)
23
+ alcohol = st.slider("Alcohol", min_value=0.0, max_value=25.0, step=0.01)
24
+
25
+ clicked = st.form_submit_button("Predict quality")
26
+
27
+ if clicked:
28
+ result=model.predict(pd.DataFrame({"fixed_acidity": [fixed_acidity],
29
+ "volatile_acidity": [volatile_acidity],
30
+ "citric_acid": [citric_acid],
31
+ "residual_sugar": [residual_sugar],
32
+ "chlorides": [chlorides],
33
+ "free_sulfur_dioxide": [free_sulfur_dioxide],
34
+ "total_sulfur_dioxide": [total_sulfur_dioxide],
35
+ "density": [density],
36
+ "ph": [ph],
37
+ "sulphates": [sulphates],
38
+ "alcohol": [alcohol]}))
39
+ predicted_quality_rank = result[0]
40
+ st.success('The predicted wine quality ranking is {}'.format(predicted_quality_rank))
41
+
42
+ if __name__=='__main__':
43
+ main()
model.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9fe062d0f6779fb8654b1914c4ff51702e89253ed6bb9a16f21e69b9af1692d
3
+ size 971676
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ joblib
2
+ pandas
3
+ scikit-learn==1.2.2
4
+ xgboost==1.7.6
5
+ altair<5