Change layout

app.py

@@ -104,8 +104,8 @@ if state.image_file is None:
     state.image = image
 
 # col1, col2 = st.beta_columns([6, 4])
-
-if st.button(
+new_col1, new_col2 = st.beta_columns([5, 5])
+if new_col1.button(
     "Get a random example",
     help="Get a random example from the 100 `seeded` image-text pairs.",
 ):
@@ -133,17 +133,19 @@ if st.button(
 
 transformed_image = get_transformed_image(state.image)
 
+
+
 # Display Image
-st.image(state.image, use_column_width="auto")
+new_col1.image(state.image, use_column_width="auto")
+
 
-new_col1, new_col2 = st.beta_columns([5, 5])
 # Display Question
-question = new_col1.text_input(
+question = new_col2.text_input(
     label="Question",
     value=state.question,
     help="Type your question regarding the image above in one of the four languages.",
 )
-new_col1.markdown(
+new_col2.markdown(
     f"""**English Translation**: {question if state.question_lang_id == "en" else translate(question, 'en')}"""
 )