Update app.py

app.py

@@ -116,15 +116,14 @@ def prediction(report_data):
         report_data=scaler.transform(report_data)
         response = model_xgb.predict(report_data)
         if response==1:
-                st.header('This customer willing to buy a vehicle insurance')
+                return 'Status of Customer. This customer willing to buy a vehicle insurance')
         else:
-                st.header('This customer will not buy a vehicle insurance')
+                return 'Status of Customer. This customer will not buy a vehicle insurance')
         
-
-
-
 y_pred = prediction(user_data)
-    
+
+if st.button("Predict"):
+    st.subheader(y_pred)