Filimize commited on
Commit
c5414fc
·
1 Parent(s): cc5555b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -13
app.py CHANGED
@@ -20,8 +20,8 @@ EDU_DICT = {'Preschool': 1,
20
  'Doctorate': 16
21
  }
22
 
23
- model = #
24
- unique_values = #
25
 
26
  unique_class = unique_values["workclass"]
27
  unique_education = unique_values["education"]
@@ -33,19 +33,19 @@ unique_race = unique_values["race"]
33
  unique_country = unique_values["native.country"]
34
 
35
  def main():
36
- st.title("Adult Income")
37
 
38
  with st.form("questionaire"):
39
- age = # user's input
40
- workclass = # user's input
41
- education = # user's input
42
- Marital_Status = # user's input
43
- occupation = # user's input
44
- relationship = # user's input
45
- race = # user's input
46
- sex = # user's input
47
- hours_per_week = # user's input
48
- native_country = # user's input
49
 
50
  # clicked==True only when the button is clicked
51
  clicked = st.form_submit_button("Predict income")
@@ -61,5 +61,9 @@ def main():
61
  "hours.per.week": [hours_per_week],
62
  "native.country": [native_country]}))
63
  # Show prediction
 
 
64
 
65
  # Run main()
 
 
 
20
  'Doctorate': 16
21
  }
22
 
23
+ model = joblib.load('model.joblib')
24
+ unique_values = joblib.load('unique_values.joblib')
25
 
26
  unique_class = unique_values["workclass"]
27
  unique_education = unique_values["education"]
 
33
  unique_country = unique_values["native.country"]
34
 
35
  def main():
36
+ st.title("Adult Income") #<-Apps' name
37
 
38
  with st.form("questionaire"):
39
+ age = st.slider('Age',min_value=10,max_value=100)
40
+ workclass = st.selectbox('work class',options=unique_class)
41
+ education = st.selectbox('education',options=unique_education)
42
+ Marital_Status = st.selectbox('marital status',options=unique_marital_status)
43
+ occupation = st.selectbox('occupation',options=unique_occupation)
44
+ relationship = st.selectbox('relationship',options=unique_relationship)
45
+ race = st.selectbox('race',options=unique_race)
46
+ sex = st.selectbox('sex',options=unique_sex)
47
+ hours_per_week = ('hours_per_week',min_value=1,max_value=100)
48
+ native_country = st.selectbox('country',options=unique_marital_country)
49
 
50
  # clicked==True only when the button is clicked
51
  clicked = st.form_submit_button("Predict income")
 
61
  "hours.per.week": [hours_per_week],
62
  "native.country": [native_country]}))
63
  # Show prediction
64
+ result ='>50k' if result[0]==1 else '<=50k'
65
+ st.success("Your prediction income is" +result)
66
 
67
  # Run main()
68
+ if __name__=="__main__":
69
+ main()