Update app.py

app.py

@@ -3,14 +3,14 @@ import joblib
 import numpy as np
 
 
-# # load model
-# loaded_model = joblib.load('model.pkl')
+# load model
+loaded_model = joblib.load('model.pkl')
 
 
-# def generate_prediction(input_array):
-#     ans = loaded_model.predict(input_array)
+def generate_prediction(input_array):
+    ans = loaded_model.predict(input_array)
 
-#     return ans
+    return ans
 
 
 # def main():
@@ -120,26 +120,32 @@ family_size = st.number_input("Family Size", min_value=1, max_value=10, value=4,
 pin = st.number_input("Pin", min_value=100000, max_value=999999, value=500000, step=1)
 
 gender = { "Male" :1,"Female" : 2}
-Gender = st.selectbox("Gender", options=list(gender.keys()))
-
+Gender_index = st.selectbox("Gender", options=list(gender.keys()))
+Gender = gender[Gender_index]
 
 Mirrage = {"Single" : 1, "Married": 2,"Not Revealed" : 3}
-Marital =  st.selectbox("Marital Status", options=list(Mirrage.keys()))
+Marital_index =  st.selectbox("Marital Status", options=list(Mirrage.keys()))
+Marital_status = Mirrage[Marital_index]
+
 
 occupation_dict = {"Student" :1, "Employee" : 2, "Self Employeed" : 3, "House wife" : 4}
-occupation = st.selectbox("Marital Status", options=list(occupation_dict.keys()))
+occupation_index = st.selectbox("Marital Status", options=list(occupation_dict.keys()))
+occupation = occupation_dict[occupation_index]
 
 educational_level = {"Graduate": 1, "Post Graduate":2, "Ph.D":3, "School" :4, "Uneducated" :5}
-educational = st.selectbox("educational_level", options=list(educational_level.keys()))
+educational_index = st.selectbox("educational_level", options=list(educational_level.keys()))
+education = educational_level[educational_index]
 
 Review_dict = {"Positive" : 1, "Negative":0}
-Review = st.multiselect("Review", options=list(Review_dict.keys()))
+Review_index = st.multiselect("Review", options=list(Review_dict.keys()))
+Review = Review_dict[Review_index]
 
 
-# # Create a button to trigger the model
-# if st.button("Predict"):
-#     # TODO: Replace with your model code
-#     prediction = "Some prediction"
+# Create a button to trigger the model
+if st.button("Predict"):
+    # TODO: Replace with your model code
+    
+    prediction = generate_prediction(np.array([[age, income, family_size, pin, Gender, Review, Marital_status, occupation, education]]))
 
-#     # Show the prediction
-#     st.write("Prediction:", prediction)
+    # Show the prediction
+    st.write("Prediction:", prediction)