Update app.py

app.py

@@ -59,7 +59,7 @@ with gr.Blocks() 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], 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, num_images_input, 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"):
@@ -77,7 +77,7 @@ with gr.Blocks() as app:
             # Update fungsi sesuai kebutuhan
             return process_image(image, prompt, neg_prompt, model)
         
-        generate_button_i2i.click(fn=process_image_func, inputs=[image_input, prompt_input_i2i, neg_prompt_input_i2i, model_dropdown], outputs=output_image_i2i)
+        generate_button_i2i.click(fn=process_image_func, inputs=[image_input, prompt_input_i2i, neg_prompt_input_i2i, model_dropdown, vae_dropdown], outputs=output_image_i2i)
 
     # Tab untuk ControlNet
     with gr.Tab("ControlNet"):
@@ -94,7 +94,7 @@ with gr.Blocks() as app:
             # Update fungsi sesuai kebutuhan
             return controlnet_process(image, controlnet_type, model)
         
-        controlnet_button.click(fn=controlnet_process_func, inputs=[controlnet_image_input, controlnet_dropdown, model_dropdown], outputs=controlnet_output_image)
+        controlnet_button.click(fn=controlnet_process_func, inputs=[controlnet_image_input, controlnet_dropdown, model_dropdown, vae_dropdown], outputs=controlnet_output_image)
 
 # Jalankan antarmuka
 app.launch()