update

app.py

@@ -95,6 +95,14 @@ def plot_indices(model, indices, lim=1.5, alpha=0.02, calculate_lim=False):
     im = plot_isovist(locs, sampled_isovists, lim, alpha, calculate_lim)
     return im
 
+st.write('''<style>
+
+[data-testid="column"] {
+    width: calc(33.3333% - 1rem) !important;
+    flex: 1 1 calc(33.3333% - 1rem) !important;
+    min-width: calc(33% - 1rem) !important;
+}
+</style>''', unsafe_allow_html=True)
 
 
 st.subheader("GIsT: Generative Isovist Transformers")
@@ -251,12 +259,3 @@ if undo:
 
 
 cont.image(st.session_state.image)
-
-st.write('''<style>
-
-[data-testid="column"] {
-    width: calc(33.3333% - 1rem) !important;
-    flex: 1 1 calc(33.3333% - 1rem) !important;
-    min-width: calc(33% - 1rem) !important;
-}
-</style>''', unsafe_allow_html=True)