Update app.py

app.py

@@ -99,7 +99,7 @@ if upload_file is not None:
                 df.columns = new_columns
                 
                 st.dataframe(df, key="csv_dataframe")
-                fig = px.treemap(df, path=[px.Constant("all"), 'df.columns', 'df.columns'],
+                fig = px.treemap(df, path=[px.Constant("all"), 'column_1', 'column_3'],
                                   color='column_2')
                 fig.update_layout(margin = dict(t=50, l=25, r=25, b=25))
                 st.subheader("Tree map", divider = "red")