Update app.py

app.py

@@ -125,6 +125,7 @@ tag_model.to(device, dtype=weight_dtype)
 def process(
     input_image: Image.Image,
     user_prompt: str,
+    use_KDS: bool,
     positive_prompt: str,
     negative_prompt: str,
     num_inference_steps: int,
@@ -175,7 +176,7 @@ def process(
                     height=height, width=width,
                     guidance_scale=cfg_scale,  conditioning_scale=1,
                     start_point='lr', start_steps=999,ram_encoder_hidden_states=ram_encoder_hidden_states,
-                    latent_tiled_size=latent_tiled_size, latent_tiled_overlap=latent_tiled_overlap
+                    latent_tiled_size=latent_tiled_size, latent_tiled_overlap=latent_tiled_overlap, use_KDS=use_KDS
                 ).images[0]
 
             if True:  # alpha<1.0:
@@ -207,6 +208,7 @@ with block:
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(type="pil")
+            use_KDS = gr.Checkbox(label="Use Kernel Density Steering")
             run_button = gr.Button("Run")
             with gr.Accordion("Options", open=True):
                 user_prompt = gr.Textbox(label="User Prompt", value="")
@@ -228,6 +230,7 @@ with block:
     inputs = [
         input_image,
         user_prompt,
+        use_KDS,
         positive_prompt,
         negative_prompt,
         num_inference_steps,