Update layout

app.py

@@ -109,9 +109,9 @@ if state.image_file is None:
     image = plt.imread(image_path)
     state.image = image
 
-col1, col2 = st.beta_columns([6, 4])
+# col1, col2 = st.beta_columns([6, 4])
 
-if col2.button(
+if st.button(
     "Get a random example",
     help="Get a random example from the 100 `seeded` image-text pairs.",
 ):
@@ -126,24 +126,21 @@ if col2.button(
     image = plt.imread(image_path)
     state.image = image
 
-col2.write("OR")
+# col2.write("OR")
 
-uploaded_file = col2.file_uploader(
-    "Upload your image",
-    type=["png", "jpg", "jpeg"],
-    help="Upload a file of your choosing.",
-)
-if uploaded_file is not None:
-    st.error(
-        "Uploading files does not work on HuggingFace spaces. This app only supports random examples for now."
-    )
-    # state.image_file = os.path.join("images/val2014", uploaded_file.name)
-    # state.image = np.array(Image.open(uploaded_file))
+# uploaded_file = col2.file_uploader(
+#     "Upload your image",
+#     type=["png", "jpg", "jpeg"],
+#     help="Upload a file of your choosing.",
+# )
+# if uploaded_file is not None:
+#     state.image_file = os.path.join("images/val2014", uploaded_file.name)
+#     state.image = np.array(Image.open(uploaded_file))
 
 transformed_image = get_transformed_image(state.image)
 
 # Display Image
-col1.image(state.image, use_column_width="auto")
+st.image(state.image, use_column_width="auto")
 
 new_col1, new_col2 = st.beta_columns([5, 5])
 # Display Question