Update app.py

app.py

@@ -126,6 +126,7 @@ def process(
     input_image: Image.Image,
     user_prompt: str,
     use_KDS: bool,
+    bandwidth: float,
     num_particles: int,
     positive_prompt: str,
     negative_prompt: str,
@@ -177,8 +178,8 @@ 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, use_KDS=use_KDS,
-                    num_particles=num_particles
+                    latent_tiled_size=latent_tiled_size, latent_tiled_overlap=latent_tiled_overlap, 
+                    use_KDS=use_KDS, bandwidth=bandwidth, num_particles=num_particles
                 ).images[0]
 
             if True:  # alpha<1.0:
@@ -211,7 +212,8 @@ with block:
         with gr.Column():
             input_image = gr.Image(type="pil")
             num_particles = gr.Slider(label="Num of Partickes", minimum=1, maximum=16, step=1, value=10)
-            use_KDS = gr.Checkbox(label="Use Kernel Density Steering")
+            bandwidth = gr.Slider(label="Bandwidth", minimum=0.1, maximum=0.8, step=0.1, value=0.1)
+            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="")
@@ -234,6 +236,7 @@ with block:
         input_image,
         user_prompt,
         use_KDS,
+        bandwidth,
         num_particles,
         positive_prompt,
         negative_prompt,