Change layout

app.py

@@ -96,15 +96,6 @@ if state.image_file is None:
 
 # col1, col2 = st.beta_columns([6, 4])
 
-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("- ")
-
-    image_path = os.path.join("images", state.image_file)
-    image = plt.imread(image_path)
-    state.image = image
-
 # col2.write("OR")
 
 # uploaded_file = col2.file_uploader("Upload your image", type=["png", "jpg", "jpeg"])
@@ -118,7 +109,15 @@ new_col1, new_col2 = st.beta_columns([5,5])
 # Display Image
 new_col1.image(state.image, use_column_width="always")
 
+if new_col2.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("- ")
 
+    image_path = os.path.join("images", state.image_file)
+    image = plt.imread(image_path)
+    state.image = image
+    
 # Display Reference Caption
 new_col2.write("**Reference Caption**: " + state.caption)
 new_col2.markdown(