Update app.py

app.py

@@ -168,11 +168,11 @@ with gr.Blocks(css=css) as demo:
                 max_lines=1,
                 placeholder="Enter your prompt",
             )
-            btn = gr.Button("Generate image").style(full_width=False)
+            btn = gr.Button("Generate image")
 
         gallery = gr.Gallery(
             label="Generated images", show_label=False, elem_id="gallery"
-        ).style(grid=[2], height="auto")
+        )
         
         with gr.Accordion("Advanced options", open=False):
             guidance = gr.Slider(