Update app.py

app.py

@@ -602,8 +602,8 @@ def display_images_and_wikipedia_summaries():
     col_sizes = ['small' if size <= 4 else 'medium' if size <= 8 else 'large' for size in grid_sizes]
 
     # Create a map for number of columns to use for each size
-    num_columns_map = {"small": 2, "medium": 3, "large": 4}
-    current_grid_size = 0
+    num_columns_map = {"small": 4, "medium": 3, "large": 2}
+    current_grid_size = 2
 
     for image_file, col_size in zip(image_files_sorted, col_sizes):
         if current_grid_size != num_columns_map[col_size]: