Update app.py

app.py

@@ -84,31 +84,31 @@ if st.button("Submit"):
         e_salary=np.round(((e_salary-100090.239881)/57510.492818),3)
         
         
-        prediction = bank_model.predict([[credit_score, geography_value, gender_value, age, tenure, balance,  e_salary,  active_holder_value]])
+        prediction = bank_model.predict([[credit_score, geography_value, gender_value, age, tenure, balance,  e_salary,  active_holder_value]])[0]
 
         st.write(prediction)
-#       # Define messages and colors
-#         review_status = {
-#             1: ("✅ The Customer is Interseted in our Bank", "#32CD32"),  # Green
-#             0: ("❌ The Customer is Not Interseted in our Bank", "#FF4500")   # Red-Orange
-#         }
+      # Define messages and colors
+        review_status = {
+            1: ("✅ The Customer is Interseted in our Bank", "#32CD32"),  # Green
+            0: ("❌ The Customer is Not Interseted in our Bank", "#FF4500")   # Red-Orange
+        }
     
-#         # Get message and color based on prediction
-#         message, color = review_status.get(prediction, ("❓ Unknown Prediction", "#808080"))
+        # Get message and color based on prediction
+        message, color = review_status.get(prediction, ("❓ Unknown Prediction", "#808080"))
     
-#         # Display styled result
-#         st.markdown(f"""
-#             <div style="
-#                 padding: 15px; 
-#                 background-color: {color}; 
-#                 border-radius: 10px; 
-#                 text-align: center; 
-#                 font-size: 18px; 
-#                 font-weight: bold; 
-#                 color: white;">
-#                 {message}
-#             </div>
-#         """, unsafe_allow_html=True)
+        # Display styled result
+        st.markdown(f"""
+            <div style="
+                padding: 15px; 
+                background-color: {color}; 
+                border-radius: 10px; 
+                text-align: center; 
+                font-size: 18px; 
+                font-weight: bold; 
+                color: white;">
+                {message}
+            </div>
+        """, unsafe_allow_html=True)
     
      except Exception as e:
          st.error(f"⚠️ Error in prediction: {e}")