Update app.py

app.py

@@ -115,16 +115,14 @@ def prediction(report_data):
         #scaling the user data
         report_data=scaler.transform(report_data)
         response = model_xgb.predict(report_data)
-        def fun():
-            if response==1:
-                st.subheader('This customer willing to buy a vehicle insurance')
+        if response==1:
+                return 'This customer willing to buy a vehicle insurance'
             else:
-                st.subheader('This customer will not buy a vehicle insurance')
-        if st.button():
-            fun()
+                return 'This customer will not buy a vehicle insurance'
+        
 
 y_pred = prediction(user_data)
-
+st.header(y_pred)