Update app.py

app.py

@@ -1,6 +1,6 @@
 import joblib
 import pandas as pd
-
+import streamlit as st
 
 EDU_DICT = {'Preschool': 1,
             '1st-4th': 2,
@@ -20,8 +20,8 @@ EDU_DICT = {'Preschool': 1,
             'Doctorate': 16
             }
 
-model = #
-unique_values = #
+model = joblib.load('model joblib')
+unique_values = joblib.load('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("Ocupation",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_country)
 
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Predict income")
@@ -61,5 +61,8 @@ 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()