ok

app.py

@@ -296,8 +296,8 @@ def create_demo(model):
             submit = gr.Button("Submit")
         
         with gr.Column():
-            design_gallery = gr.Gallery(label="Output Gallery", elem_id='img-display-output').style(grid=[2], height='auto')
-    
+            design_gallery = gr.Gallery(label="Output Gallery", elem_id='img-display-output')
+
     def on_submit(image, text, num_steps, guidance_scale, seed, strength, a_prompt, n_prompt, img_size):
         model.seed = seed
         model.neg_prompt = n_prompt