Fix layout

apps/mlm.py

@@ -67,7 +67,12 @@ def app(state):
         with st.spinner("Loading model..."):
             mlm_state.mlm_model = load_model(mlm_checkpoints[0])
 
-    col1, col2 = st.beta_columns([1,1])
+    query1 = st.text_input(
+        "Enter a URL to an image",
+        value="http://images.cocodataset.org/val2017/000000039769.jpg",
+    )
+
+    col1, col2, col3 = st.beta_columns([2,1, 2])
     if col1.button(
         "Get a random example",
         help="Get a random example from the 100 `seeded` image-text pairs.",
@@ -87,13 +92,9 @@ def app(state):
         image = plt.imread(image_path)
         mlm_state.mlm_image = image
 
-    st.write("OR")
-    
-    query1 = st.text_input(
-        "Enter a URL to an image",
-        value="http://images.cocodataset.org/val2017/000000039769.jpg",
-    )
-    if col2.button("Use above URL"):
+    col2.write("OR")
+
+    if col3.button("Use above URL"):
         image_data = requests.get(query1, stream=True).raw
         image = np.asarray(Image.open(image_data))
         mlm_state.mlm_image = image

apps/vqa.py

@@ -76,8 +76,11 @@ def app(state):
             vqa_state.vqa_model = load_model(vqa_checkpoints[0])
     
     # Display Top-5 Predictions
-    
-    col1, col2 = st.beta_columns([1,1])
+    query1 = st.text_input(
+        "Enter a URL to an image",
+        value="http://images.cocodataset.org/val2017/000000039769.jpg",
+    )
+    col1, col2, col3 = st.beta_columns([2,1, 2])
     if col1.button(
         "Get a random example",
         help="Get a random example from the 100 `seeded` image-text pairs.",
@@ -93,13 +96,9 @@ def app(state):
         image = plt.imread(image_path)
         vqa_state.vqa_image = image
 
-    st.write("OR")
+    col2.write("OR")
     
-    query1 = st.text_input(
-        "Enter a URL to an image",
-        value="http://images.cocodataset.org/val2017/000000039769.jpg",
-    )
-    if col2.button("Use above URL"):
+    if col3.button("Use above URL"):
         image_data = requests.get(query1, stream=True).raw
         image = np.asarray(Image.open(image_data))
         vqa_state.mlm_image = image