Change Layout

app.py

@@ -104,8 +104,8 @@ if state.image_file is None:
     state.image = image
 
 # col1, col2 = st.beta_columns([6, 4])
-new_col1, new_col2 = st.beta_columns([5, 5])
-if new_col1.button(
+
+if st.button(
     "Get a random example",
     help="Get a random example from the 100 `seeded` image-text pairs.",
 ):
@@ -133,7 +133,7 @@ if new_col1.button(
 
 transformed_image = get_transformed_image(state.image)
 
-
+new_col1, new_col2 = st.beta_columns([5, 5])
 
 # Display Image
 new_col1.image(state.image, use_column_width="auto")