Make section height responsive to content for filterable gallery?

I am using a custom code block to create a filterable gallery of a bunch of images. By default, it displays all +20 images, but when I click one of the categories to filter the images (e.g., by kitchen, by living room, etc.) it cuts down to just a handful, the section height looks to remain the same. This creates a ton of white space in the middle of the site. How might I get this section to respond to the height of the content being presented based on the filtered view? 

Here's a before click: 

Here's after

I've been using openai to help write the code, but I'm stuck on this. Any ideas? 

Hi,

Can you share link to this page?