Update app.py

app.py

@@ -90,7 +90,8 @@ with gr.Blocks(css=css) as demo:
             )
             diffuse_btn = gr.Button(value="🏭 Render Images for My Prompts")
         with gr.Column(elem_id="generated-gallery"):
-            sd_output = gr.Gallery().style(grid=2, height="auto")
+            # Pass styling parameters directly to gr.Gallery
+            sd_output = gr.Gallery(columns=2, height="auto")
             with gr.Group(elem_id="share-btn-container"):
                 # If you have HTML content for these icons, you can include it here
                 community_icon = gr.HTML("", visible=False)