Update app.py

app.py

@@ -276,8 +276,8 @@ def main():
             if st.sidebar.checkbox("Neural Network"):
                 st.subheader(":orange[Neural Network] :sunglasses:")
                 ann = load_model("ann.h5")
-                pred = ann.predict(X)
-                pred_proba = ann.predict_proba(X)
+                pred_proba = ann.predict(X)
+                pred = np.where(pred_proba < 0.5, 0, 1)
                 st.subheader(':green[Prediction]')
                 loan_status = np.array(['N','Y'])
                 prediction = pd.DataFrame(loan_status[pred], columns=['prediction'])
@@ -422,12 +422,12 @@ def main():
             if st.sidebar.checkbox("Neural Network"):
                 st.subheader(":orange[Neural Network]")
                 ann = load_model('ann.h5')
-                pred = ann.predict(data_input)
+                pred_proba = ann.predict(data_input)
                 if pred == 1:
                     st.write(":orange[__Le pret peut etre octroyer__] :white_check_mark:")
                 else: 
                     st.write(":red[__Desole,...__] :disappointed:")
-                pred_proba = ann.predict_proba(data_input)
+                pred = np.where(pred_proba < 0.5, 0, 1)
                 loan_status = np.array(['N','Y'])
                 prediction = pd.DataFrame(loan_status[pred], columns=['prediction'])
                 df = pd.concat([data_input, prediction], axis=1)