Update app.py

app.py

@@ -24,46 +24,33 @@ remote_ratio_dict = {'0':1,
 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_native_country =  unique_values["native.country"]
+unique_experience_level =  unique_values["experience_level"]
+unique_employment_type =  unique_values["employment_type"]
+unique_job_title =  unique_values["job_title"]
+unique_remote_ratio =  unique_values["remote_ratio"]
+unique_company_size =  unique_values["company_size"]
 
 def main():
-    st.title("Adult Income")
+    st.title("Data Science Job Salaries")
 
     with st.form("questionaire"):
-        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 = 10, max_value = 100)
-        native_country = st.selectbox("Native country", options=unique_native_country)
+        experience_level = st.selectbox("The experience level in the job", options = unique_experience_level)
+        employment_type = st.selectbox("The type of employement for the role", options=unique_employment_type)
+        job_title = st.selectbox("The role worked", options=unique_job_title)
+        remote_ratio = st.selectbox("The overall amount of work done remotely", options=unique_remote_ratio)
+        company_size = st.selectbox("The average number of people that worked for the company", options=unique_company_size)
 
         # clicked==True only when the button is clicked
-        clicked = st.form_submit_button("Predict income")
+        clicked = st.form_submit_button("Predict Data Science Job Salaries")
         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]}))
+            result=model.predict(pd.DataFrame({"experience_level": [Experience_DICT[experience_level],
+                                               "employment_type": [employment_type_dict[employment_type]],
+                                               "job_title": [job_title],
+                                               "remote_ratio": [remote_ratio],
+                                               "company_size": [Company_size[company_size]]}))
             # Show prediction
-            result = ">50K" if result[0] == 1 else "<=50K"
-            st.success("Your predicted income is " +result)
+            salary_in_usd = str(round(float(result),2))
+            st.success("Your predicted Data Science job salaries in usd is " +result)
             
 if __name__ == "__main__":
     main()