Update app.py

app.py

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