Update app.py

app.py

@@ -376,13 +376,13 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Row():
         with gr.Column(scale=55):
-            prompt = gr.Textbox(label="Prompt", show_label=False, max_lines=2, placeholder="Enter your prompt").style(container=False)
-            negative_prompt = gr.Textbox(label="Negative Prompt", show_label=False, max_lines=2, placeholder="Enter negative prompt").style(container=False)
+            prompt = gr.Textbox(label="Prompt", show_label=False, max_lines=2, placeholder="Enter your prompt")
+            negative_prompt = gr.Textbox(label="Negative Prompt", show_label=False, max_lines=2, placeholder="Enter negative prompt")
         with gr.Column(scale=45):
             generate_btn = gr.Button("Generate")
 
     with gr.Row():
-        image_output = gr.Gallery(label="Generated Images").style(grid=2, height="auto")
+        image_output = gr.Gallery(label="Generated Images").style(grid=[2], height="auto")
         seed_output = gr.Number(label="Seed Used")
 
     gr.Examples(examples=examples, inputs=[prompt])
@@ -395,4 +395,4 @@ with gr.Blocks(css=css) as demo:
 
     generate_btn.click(fn=generate, inputs=inputs, outputs=[image_output, seed_output])
 
-demo.launch()
+demo.launch()