no number of rows control... it's not changing anything

app.py

@@ -77,8 +77,8 @@ def get_images(batch_size, only_rgb, state):
 
     return images, DataFrame(metadatas)
 
-def update_shape(rows, columns):
-    return gr.update(rows=rows, columns=columns)
+def update_shape(columns):
+    return gr.update(columns=columns)
 
 def new_state():
     return gr.State({})
@@ -97,7 +97,7 @@ if __name__ == "__main__":
             label=ev.DATASET,
             interactive=False,
             object_fit="scale-down",
-            columns=5, rows=2, render=False)
+            columns=5, render=False)
 
         with gr.Row():
             dataset = gr.Textbox(label="Dataset", value=ev.DATASET, interactive=False)
@@ -116,11 +116,9 @@ if __name__ == "__main__":
         with gr.Row():
             batch_size.render()
 
-            rows = gr.Number(2, label="Rows")
             columns = gr.Number(5, label="Columns")
 
-            rows.change(update_shape, [rows, columns], [gallery])
-            columns.change(update_shape, [rows, columns], [gallery])
+            columns.change(update_shape, [columns], [gallery])
 
         with gr.Row():
             shard.render()