Update app.py

app.py

@@ -1,5 +1,6 @@
 import joblib
 import pandas as pd
+import streamlit as st
 
 
 EDU_DICT = {'Preschool': 1,
@@ -20,8 +21,8 @@ EDU_DICT = {'Preschool': 1,
             'Doctorate': 16
             }
 
-model = #
-unique_values = #
+model = joblib.load('model.joblib')
+unique_values = loblib.load('unique_values.loblib')
     
 unique_class =  unique_values["workclass"]
 unique_education =  unique_values["education"]
@@ -30,22 +31,22 @@ unique_relationship =  unique_values["relationship"]
 unique_occupation =  unique_values["occupation"]
 unique_sex =  unique_values["sex"]
 unique_race = unique_values["race"]
-unique_country =  unique_values["native.country"]
+unique_native_country =  unique_values["native.country"]
 
 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_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 = 10, 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 +62,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)
+            
+if __name__ = "__main__":
+    main()
 
-# Run main()