Chatop commited on
Commit
f41fb3c
·
1 Parent(s): 571c52f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -13
app.py CHANGED
@@ -31,22 +31,22 @@ unique_relationship = unique_values["relationship"]
31
  unique_occupation = unique_values["occupation"]
32
  unique_sex = unique_values["sex"]
33
  unique_race = unique_values["race"]
34
- unique_country = unique_values["native.country"]
35
 
36
  def main():
37
  st.title("Adult Income")
38
 
39
  with st.form("questionaire"):
40
- age = # user's input
41
- workclass = # user's input
42
- education = # user's input
43
- Marital_Status = # user's input
44
- occupation = # user's input
45
- relationship = # user's input
46
- race = # user's input
47
- sex = # user's input
48
- hours_per_week = # user's input
49
- native_country = # user's input
50
 
51
  # clicked==True only when the button is clicked
52
  clicked = st.form_submit_button("Predict income")
@@ -62,5 +62,8 @@ def main():
62
  "hours.per.week": [hours_per_week],
63
  "native.country": [native_country]}))
64
  # Show prediction
65
-
66
- # Run main()
 
 
 
 
31
  unique_occupation = unique_values["occupation"]
32
  unique_sex = unique_values["sex"]
33
  unique_race = unique_values["race"]
34
+ unique_country= unique_values["native.country"]
35
 
36
  def main():
37
  st.title("Adult Income")
38
 
39
  with st.form("questionaire"):
40
+ age = st.slider("Age", min_value=10, max_value=100)
41
+ workclass = st.selectbox("WorkClass", options=unique_class)
42
+ education = st.selectbox("Education", options=unique_education )
43
+ Marital_Status = st.selectbox("Marital Status ", options=unique_marital_status )
44
+ occupation = st.selectbox("Occupation ", options=unique_occupation )
45
+ relationship = st.selectbox("Relationship ", options=unique_relationship )
46
+ race = st.selectbox("Race ", options=unique_race)
47
+ sex = st.selectbox("Sex", options=unique_sex)
48
+ hours_per_week = st.slider("Hours per week", min_value=1, max_value=100)
49
+ native_country = st.selectbox("Native country ", options=unique_country)
50
 
51
  # clicked==True only when the button is clicked
52
  clicked = st.form_submit_button("Predict income")
 
62
  "hours.per.week": [hours_per_week],
63
  "native.country": [native_country]}))
64
  # Show prediction
65
+ result = '>50k' if result[0] == 1 else 0 '<=50k'
66
+ st.success("Your predicted income is "+result)
67
+
68
+ if __name__ == "__main__":
69
+ main()