Update app.py

app.py

@@ -3,45 +3,45 @@ import pandas as pd
 import streamlit as st
 
 model = joblib.load('model.joblib')
-unique_values = #
+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_area_type =  unique_values["Area Type"]
+unique_city =  unique_values["City"]
+unique_furnishing =  unique_values["Furnishing Status"]
+unique_tenant =  unique_values["Tenant Preferred"]
+unique_contact =  unique_values["Point of Contact"]
+
 
 def main():
-    st.title("Adult Income")
+    st.title("House Rent Prediction")
 
     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
+        BHK = st.slider("BHK",min_value=1,max_value=6)
+        Size = st.slider("Size",min_value=10,max_value=8000)
+        Bathroom = st.slider("Bathroom",min_value=1,max_value=10)
+        FloorLevels = st.slider("Floor Level",min_value=1,max_value=89)
+        TotalFloor = st.slider("Total Floors",min_value=1,max_value=89)
+        AreaType = st.selectbox("Area Type",options=unique_area_type)
+        city = st.selectbox("City",options=unique_city)
+        FurnishingStatus = st.selectbox("Furnishing Status",options=unique_furnishing)
+        TenantPreferred = st.selectbox("Tenant Preferred",options=unique_tenant)
+        PointofContact = st.selectbox("Point of Contact",options=unique_contact)
 
         # 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({"BHK": [BHK],
+                                               "Size": [Size],
+                                               "Bathroom": [Bathroom],
+                                               "Floor Level": [FloorLevels],
+                                               "Total Floors": [TotalFloor],
+                                               "Area Type": [AreaType],
+                                               "City": [city],
+                                               "Furnishing Status": [FurnishingStatus],
+                                               "Tenant Preferred": [TenantPreferred],
+                                               "Point of Contact": [PointofContact]}))
             # Show prediction
+            st.success('Your predicted rent is'+result)
 
-# Run main()
+if __name__ == "__main__":
+    main()