Update app.py

app.py

@@ -2,28 +2,19 @@ import joblib
 import pandas as pd
 import streamlit as st
 
-
-model = joblib.load('model.joblib')
-unique_values = joblib.load('unique_values.joblib')
-
-EDU_DICT = {'Partial High School': 1,
-            'High School': 2,
-            'Partial College' : 3,
-            'Bachelors' : 4,
-            'Graduate Degree' : 5}
-
 dist_dict = {'0-1 Miles' : 1,
              '1-2 Miles' : 2,
              '2-5 Miles' : 3,
              '5-10 Miles' : 4,
              '10+ Miles' : 5}
-    
+
+model = joblib.load('model.joblib')
+unique_values = joblib.load('unique_values.joblib')
+
 unique_Marital_Status =  unique_values["Marital Status"]
 unique_Gender =  unique_values["Gender"]
-unique_Occupation =  unique_values["Occupation"]
 unique_Home_Owner =  unique_values["Home Owner"]
 unique_Region =  unique_values["Region"]
-unique_Education =  unique_values["Education"]
 unique_Commute_Distance =  unique_values["Commute Distance"]
 
 
@@ -33,9 +24,8 @@ def main():
     with st.form("questionaire"):
         Marital_Status = st.selectbox("Marital Status", options=unique_Marital_Status)
         Gender = st.selectbox("Gender", options=unique_Gender)
-        Education = st.selectbox("Education", options=unique_Education)
-        Occupation = st.selectbox("Occupation", options=unique_Occupation)
         Home_Owner = st.selectbox("Home Owner", options=unique_Home_Owner)
+        Region = st.selectbox("Region", options=unique_Region)
         Commute_Distance = st.selectbox("Commute Distance", options=unique_Commute_Distance)
         Age = st.slider("Age", min_value=0, max_value=100)
 
@@ -45,9 +35,8 @@ def main():
         if clicked:
             result=model.predict(pd.DataFrame({"Marital Status": [Marital_Status],
                                                "Gender": [Gender],
-                                               "Education": [Education],
-                                               "Occupation": [Occupation],
                                                "Home Owner": [Home_Owner],
+                                               "Region" : [Region]
                                                "Commute Distance": [Commute_Distance],
                                                "Age": [Age]}))
             # Show prediction