Update app.py

app.py

@@ -48,7 +48,9 @@ with gr.Blocks(css= "style.css") as app:
                 height_input = gr.Slider(minimum=128, maximum=2048, step=128, label="Height", value=512)
                 cfg_scale_input = gr.Slider(minimum=1, maximum=20, step=1, label="CFG Scale", value=7)
                 seed_input = gr.Number(label="Seed", value=-1)
-                num_images_input = gr.Slider(minimum=1, maximum=10, step=1, label="Batch size", value=1)
+                batch_size = gr.Slider(minimum=1, maximum=10, step=1, label="Batch size", value=1)
+                batch_count = gr.Slider(minimum=1, maximum=10, step=1, label="Batch size", value=1)
+            
             
             with gr.Column():
                 # Gallery untuk output gambar
@@ -60,7 +62,7 @@ with gr.Blocks(css= "style.css") as app:
             # Update fungsi sesuai kebutuhan
             return generate_image(prompt, neg_prompt, width, height, scheduler, num_steps, num_images, cfg_scale, seed, model)
         
-        generate_button.click(fn=update_images, inputs=[prompt_input, neg_prompt_input, width_input, height_input, scheduler_input, num_steps_input, num_images_input, cfg_scale_input, seed_input, model_dropdown, vae_dropdown], outputs=[output_gallery, output_text])
+        generate_button.click(fn=update_images, inputs=[prompt_input, neg_prompt_input, width_input, height_input, scheduler_input, num_steps_input, batch_size, batch_count, cfg_scale_input, seed_input, model_dropdown, vae_dropdown], outputs=[output_gallery, output_text])
 
     # Tab untuk Image-to-Image
     with gr.Tab("Image-to-Image"):