Update app layout

app.py

@@ -91,7 +91,7 @@ if state.model is None:
     with st.spinner("Loading model..."):
         state.model = load_model(checkpoints[0])
 
-first_index = 25
+first_index = 30
 # Init Session State
 if state.image_file is None:
     state.image_file = dummy_data.loc[first_index, "image_file"]
@@ -103,7 +103,7 @@ if state.image_file is None:
 
 new_col1, new_col2 = st.beta_columns([5,5])
 
-if new_col2.button("Get a random example", help="Get a random example from one of the seeded examples."):
+if new_col1.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("- ")
@@ -117,10 +117,11 @@ transformed_image = get_transformed_image(state.image)
 new_col1.image(state.image, use_column_width="always")
 
 # Display Reference Caption
-new_col2.write("**Reference Caption**: " + state.caption)
-new_col2.markdown(
-    f"""**English Translation**: {translate(state.caption, 'en')}"""
-)
+with new_col1.beta_expander("Reference Caption"):
+    st.write("**Reference Caption**: " + state.caption)
+    st.markdown(
+        f"""**English Translation**: {translate(state.caption, 'en')}"""
+    )
 
 
 sequence = ['']
@@ -130,10 +131,10 @@ if new_col2.button("Generate Caption", help="Generate a caption in the specified
 # print(sequence)
 
 if sequence!=['']:
-    st.write(
+    new_col2.write(
         "**Generated Caption**: "+sequence[0]
     )
 
-    st.write(
+    new_col2.write(
         "**English Translation**: "+  translate(sequence[0])
     )