Update app.py

app.py

@@ -93,7 +93,8 @@ with gr.Blocks() as demo:
         model_dropdown = gr.Dropdown(choices=MODEL_NAMES, label="Select Model(s)", multiselect=True, value=MODEL_NAMES[:3])
     
     generate_btn = gr.Button("Generate Heatmaps")
-    gallery = gr.Gallery(label="Heatmaps").style(grid=[2], height="auto")
+    # Use the 'columns' parameter to set a grid layout in the gallery.
+    gallery = gr.Gallery(label="Heatmaps", columns=2)
     
     generate_btn.click(fn=generate_heatmaps, inputs=model_dropdown, outputs=gallery)