app.py

import joblib
import pandas as pd
import streamlit as st #<-

EDU_DICT = {'Preschool': 1,
            '1st-4th': 2,
            '5th-6th': 3,
            '7th-8th': 4,
            '9th': 5,
            '10th': 6,
            '11th': 7,
            '12th': 8,
            'HS-grad': 9, 
            'Some-college': 10,
            'Assoc-voc': 11,
            'Assoc-acdm': 12,
            'Bachelors': 13,
            'Masters': 14,
            'Prof-school': 15,
            'Doctorate': 16
            }

model = joblib.load('model.joblib') #<-
unique_values = joblib.load('unique_values.joblib') #<-
    
unique_class =  unique_values["workclass"]
unique_education =  unique_values["education"]
unique_marital_status =  unique_values["marital.status"]
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"]

def main(): #<- defined main
    st.title("Adult Income") #<- app name

    with st.form("questionaire"): #<- input from user 
        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_country) #<-

        # clicked==True only when the button is clicked
        clicked = st.form_submit_button("Predict income")
        if clicked:
            result=model.predict(pd.DataFrame({"age": [age],
                                               "workclass": [workclass],
                                               "education": [EDU_DICT[education]],
                                               "marital.status": [Marital_Status],
                                               "occupation": [occupation],
                                               "relationship": [relationship],
                                               "race": [race],
                                               "sex": [sex],
                                               "hours.per.week": [hours_per_week],
                                               "native.country": [native_country]}))
            # Show prediction
            result= '>50k' if result[0]==1 else '<=50k'
            #print report 
            st.success("Your predictor income is "+result)
            
# Run main() 
#when import but dont wanna run main so..
if __name__== "__main__":
    main()