Update app.py

app.py

@@ -20,8 +20,8 @@ EDU_DICT = {'Preschool': 1,
             'Doctorate': 16
             }
 
-model = #
-unique_values = #
+model = joblib.load(model,'model.joblib')
+unique_values = joblib.dump(unique_values,'unique_values.joblib')
     
 unique_class =  unique_values["workclass"]
 unique_education =  unique_values["education"]
@@ -36,16 +36,16 @@ def main():
     st.title("Adult Income")
 
     with st.form("questionaire"):
-        age = # user's input
-        workclass = # user's input
-        education = # user's input
-        Marital_Status = # user's input
-        occupation = # user's input
-        relationship = # user's input
-        race = # user's input
-        sex = # user's input
-        hours_per_week = # user's input
-        native_country = # user's input
+        age = st.slider("Age" , min_value=10 , max_value=100)
+        workclass = st.selectbox("Workclass" , options=unique_class)
+        education = st.selectbox("Education" , options=unique_education)
+        Marital_Status = st.selectbox("Marital status" , options=unique_marital_status)
+        occupation = st.selectbox("Occupation" , options=unique_occupation)
+        relationship = st.selectbox("Relationship" , options=unique_relationship)
+        race = st.selectbox("Race" , options=unique_race)
+        sex = st.selectbox("Sex" , options=unique_sex)
+        hours_per_week = st.slider("Hours per week" , min_value=1 , max_value=100)
+        native_country = st.selectbox("Native_country" , options=unique_native_country)
 
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Predict income")
@@ -61,5 +61,9 @@ def main():
                                                "hours.per.week": [hours_per_week],
                                                "native.country": [native_country]}))
             # Show prediction
-
+            result = '>50k' if result[0] == 1 else '<=50k'
+            st.success("Your predicted income is "+result)
+            
 # Run main()
+if __name__ = "__main__" :
+    main()