Update app.py

app.py

@@ -317,7 +317,10 @@ with gr.Blocks() as demo:
                          "a waterfall in Iceland with a rainbow"
                         ], inputs=desc)
             search_btn = gr.Button("Find Images", scale = 1)
-        gallery = gr.Gallery(show_label=False, grid=(2,2,3,5))
+        gallery = gr.Gallery(
+            show_label=False,
+            columns=[2, 3, 5],
+            height="500px")
         search_btn.click(search,inputs=desc, outputs=gallery, postprocess=False)
         gr.HTML(f"Dataset: <a href='https://github.com/unsplash/datasets' target='_blank'>Unsplash Lite</a>; Number of Images: {dataset_size}")