Update app.py

app.py

@@ -103,6 +103,7 @@ def main():
 
             all_columns = insurance_claims_reduced.columns.tolist()
             selected_columns = st.multiselect("Choose columns", all_columns, default=all_columns)
+            insurance_claims_reduced = insurance_claims_reduced[selected_columns].copy()
             
             st.header("Inference Description")
             insurance_claims_reduced.describe().T
@@ -124,7 +125,7 @@ def main():
             st.pyplot(fig)
 
             if st.button("Prediction"):
-                insurance_claims_reduced = insurance_claims_reduced[selected_columns].copy()
+                #insurance_claims_reduced = insurance_claims_reduced[selected_columns].copy()
                 
                 s = setup(insurance_claims_reduced, session_id = 123, remove_multicollinearity=p_remove_multicollinearity, multicollinearity_threshold=p_multicollinearity_threshold,
                         # remove_outliers=p_remove_outliers, outliers_method=p_outliers_method,