Update app

app.py

@@ -103,10 +103,9 @@ if state.image_file is None:
     image = plt.imread(image_path)
     state.image = image
 
-col1, col2, col3 = st.beta_columns([3,3,3])
-col1.write("")
-col3.write("")
-if col2.button(
+# col1, col2, col3 = st.beta_columns([3,3,3])
+
+if st.button(
     "Get a random example",
     help="Get a random example from the 100 `seeded` image-text pairs.",
 ):

resize_images.py

@@ -2,7 +2,7 @@ import cv2
 import os
 
 
-# A function to resize the images with longer side to 500 pixels with cubic interpolation and save to a new directory
+# A function to resize the images with longer side to 300 pixels with cubic interpolation and save to a new directory
 def resize_images(path, new_path):
     if not os.path.exists(new_path):
         os.makedirs(new_path)
@@ -11,10 +11,10 @@ def resize_images(path, new_path):
             img = cv2.imread(os.path.join(path, filename))
             height, width, channels = img.shape
             if height > width:
-                new_height = 500
+                new_height = 300
                 new_width = int(width * new_height / height)
             else:
-                new_width = 500
+                new_width = 300
                 new_height = int(height * new_width / width)
             img = cv2.resize(img, (new_width, new_height), interpolation=cv2.INTER_CUBIC)
             cv2.imwrite(os.path.join(new_path, filename), img)