Update layout

app.py

@@ -117,10 +117,7 @@ if state.image_file is None:
     image = plt.imread(image_path)
     state.image = image
 
-new_col1, new_col2 = st.beta_columns([5,5])
-
-
-if new_col1.button("Get a random example", help="Get a random example from one of the seeded examples."):
+if st.button("Get a random example", help="Get a random example from one of the seeded examples."):
     sample = dummy_data.sample(1).reset_index()
     state.image_file = sample.loc[0, "image_file"]
     state.caption = sample.loc[0, "caption"].strip("- ")
@@ -132,6 +129,8 @@ if new_col1.button("Get a random example", help="Get a random example from one o
 
 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="always")
 # Display Reference Caption