Update app.py

app.py

@@ -178,7 +178,7 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                     outputs=[result, seed],
                 )
                 
-        text_button.click(result, inputs=[custom_lora, text_prompt, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=gr.Gallery[image_output, seed_output])
+        text_button.click(result, inputs=[custom_lora, text_prompt, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=[image_output, num_images_per_prompt, seed_output])
     
         with gr.Tab("Flip Image"):
             with gr.Row():