Update app.py

app.py

@@ -296,7 +296,7 @@ def create_demo(model):
             submit = gr.Button("Submit")
         
         with gr.Column():
-            design_gallery = gr.Gallery(label="Output Gallery", elem_id='img-display-output')
+            design_gallery = gr.Gallery(label="Output Gallery", elem_id='img-display-output', elem_id_css={'height': '600px', 'width': '800px'})
 
     def on_submit(image, text, num_steps, guidance_scale, seed, strength, a_prompt, n_prompt, img_size):
         model.seed = seed