better gallery css

app.py

@@ -93,6 +93,18 @@ tr:has(> td div span span div.diffDeletion) {
 tr td {
     border-top: 1px solid black;
 }
+.grid-container {
+    gap: 0;
+    grid-template-rows: auto;
+    grid-auto-rows: auto;
+}
+.thumbnail-item {
+    aspect-ratio: auto;
+    height: min-content;
+}
+.grid-wrap {
+    min-height: 0;
+}
 """
 
 
@@ -124,7 +136,6 @@ with gr.Blocks(css=css, js=make_gallery_image_buttons_js) as demo:
         show_download_button=False,
         show_fullscreen_button=False,
         elem_id="pipeline-gallery",
-        height="200px",
         allow_preview=False,
     )
     gallery_image_buttons = [gr.Button(visible=False, elem_classes="block-button") for _ in blocks]  # hack to simulate each image galery as a button, see `make_gallery_image_buttons_js``