Update app.py

app.py

@@ -1,65 +1,72 @@
 import joblib
 import pandas as pd
+import streamlit as st
 
+buy_dict = {'low': 1,
+            'med': 2,
+            'high': 3,
+            'vhigh': 4}
 
-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
-            }
+maint_dict = {'low': 1,
+            'med': 2,
+            'high': 3,
+            'vhigh': 4}
 
-model = #
-unique_values = #
+lug_dict = {'small': 1,
+            'med': 2,
+            'big': 3}
+
+safety_dict = {'low': 1,
+               'med': 2,
+               'high': 3}
+
+class_dict = {'unacc': 0,
+              'acc': 1,
+              'good': 2,
+              'vgood': 3}
+
+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"]
+unique_buy =  unique_values["buying"]
+unique_maint =  unique_values["maintenance"]
+unique_door =  unique_values["doors"]
+unique_person =  unique_values["persons"]
+unique_lugg =  unique_values["luggage_boot"]
+unique_safety =  unique_values["safety"]
 
 def main():
-    st.title("Adult Income")
+    st.title("Car Evaluation")
 
     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
+        buy = st.selectbox('Buying Price', unique_buy)
+        maint = st.selectbox('Maintenance cost', unique_maint)
+        door = st.selectbox('Door', unique_door)
+        person = st.selectbox('Persons capacity', unique_person)
+        lugg = st.selectbox('Size of luggage boot', unique_lugg)
+        safety = st.selectbox('Estimated safety of the car', unique_safety)
 
         # 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]}))
+            result=model.predict(pd.DataFrame({"buying": [buy_dict[buy]],
+                                               "maintenance": [maint_dict[maint]],
+                                               "doors": [door],
+                                               "persons": [person],
+                                               "luggage_boot": [lug_dict[lugg]],
+                                               "safety": [safety_dict[safety]]}))
             # Show prediction
+            if result[0] == 0:
+                result = 'Unacceptable'
+            elif result[0] == 1:
+                result = 'Acceptable'
+            elif result[0] == 2:
+                result = 'Good'
+            elif result[0] == 3:
+                result = 'Very Good'
+            else:
+                result = 'ERROR'
+            st.success('Your predicted car evaluation is '+result)
 
-# Run main()
+if __name__ == '__main__':
+    main() # Run main()