Update app.py

app.py

@@ -126,6 +126,7 @@ def process(
     input_image: Image.Image,
     user_prompt: str,
     use_KDS: bool,
+    num_particles: int,
     positive_prompt: str,
     negative_prompt: str,
     num_inference_steps: int,
@@ -176,7 +177,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
+                    latent_tiled_size=latent_tiled_size, latent_tiled_overlap=latent_tiled_overlap, use_KDS=use_KDS,
+                    num_particles=num_particles
                 ).images[0]
 
             if True:  # alpha<1.0:
@@ -208,6 +210,7 @@ with block:
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(type="pil")
+            num_particles = gr.Slider(label="Num of Partickes", minimum=1, maximum=16, step=1, value=4)
             use_KDS = gr.Checkbox(label="Use Kernel Density Steering")
             run_button = gr.Button("Run")
             with gr.Accordion("Options", open=True):
@@ -231,6 +234,7 @@ with block:
         input_image,
         user_prompt,
         use_KDS,
+        num_particles,
         positive_prompt,
         negative_prompt,
         num_inference_steps,