fix more

app.py

@@ -371,11 +371,11 @@ with block:
                 </h1>
               </div>
               <p style="margin-bottom: 10px; font-size: 94%; line-height: 23px;">
-                Stable Diffusion 2.1 Demo App. <br />
+                small stable diffusion Demo App. <br />
                 Click <strong>Generate image</strong> Button to generate image. <br />
                 Also Change params to have a try <br />
-                512*512 is optimized, every image will cost 30s. <br />
-                other size may cost more time. <br />
+                every 512*512 image will cost about 40s. <br />
+                more size may cost more time. <br />
                 It's just a simplified demo, you can use more advanced features optimize image quality <br />
               </p>
             </div>
@@ -439,9 +439,9 @@ with block:
                             full_width=False,
                         )
 
-                gallery = gr.Gallery(
+                gallery = gr.Image(
                     label="Generated images", show_label=False, elem_id="gallery"
-                ).style(grid=[2], height="auto")
+                ).style(height=512)
 
         # with gr.Group(elem_id="container-advanced-btns"):
         #     # advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")