Update app.py

app.py

@@ -100,6 +100,9 @@ def main():
             num_rows = int(insurance_claims.shape[0]*int(num_lines)/100)
             insurance_claims_reduced = insurance_claims.head(num_rows)
             st.write("Rows to be processed: " + str(num_rows))
+
+            all_columns = insurance_claims_reduced.columns.tolist()
+            selected_columns = st.multiselect("Choose columns", all_columns, default=all_columns)
             
             st.header("Inference Description")
             insurance_claims_reduced.describe().T
@@ -120,9 +123,6 @@ def main():
             # Display the heatmap in Streamlit
             st.pyplot(fig)
 
-            all_columns = insurance_claims_reduced.columns.tolist()
-            selected_columns = st.multiselect("Choose columns", all_columns, default=all_columns)
-
             if st.button("Prediction"):
                 insurance_claims_reduced = insurance_claims_reduced[selected_columns].copy()