Update app.py

app.py

@@ -192,6 +192,10 @@ tag_model.to(device, dtype=weight_dtype)
 def process(
     input_image: Image.Image,
     user_prompt: str,
+    use_KDS: bool,
+    bandwidth: float,
+    patch_size: int,
+    num_particles: int,
     positive_prompt: str,
     negative_prompt: str,
     num_inference_steps: int,
@@ -208,15 +212,13 @@ def process(
     ])
 
 
-    prompt_tag = _generate_vlm_prompt(
+    user_prompt = _generate_vlm_prompt(
                 vlm_model=vlm_model,
                 vlm_processor=vlm_processor,
                 process_vision_info=process_vision_info,
                 pil_image=input_image,
                 device=device,
             )
-
-    print(f'oh lala, prompt tag:{prompt_tag}')
     
     # with torch.no_grad():
     seed_everything(seed)
@@ -253,7 +255,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,
+                    latent_tiled_size=latent_tiled_size, latent_tiled_overlap=latent_tiled_overlap, 
+                    use_KDS=use_KDS, bandwidth=bandwidth, num_particles=num_particles, patch_size=patch_size,
                 ).images[0]
 
             if True:  # alpha<1.0:
@@ -312,9 +315,12 @@ with block:
             examples = gr.Examples(
                 examples=[
                     [
-                        "preset/datasets/test_datasets/179.png", 
+                        "preset/datasets/test_datasets/man.png", 
                         "",
                         False,
+                        0.1,
+                        4,
+                        4,
                         "clean, high-resolution, 8k, best quality, masterpiece",
                         "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
                         50,
@@ -324,11 +330,14 @@ with block:
                         320,
                         4,
                         1,
-                    ],    
+                    ],
                     [
-                        "preset/datasets/test_datasets/apologise.png", 
+                        "preset/datasets/test_datasets/man.png", 
                         "",
-                        False,
+                        True,
+                        0.1,
+                        16,
+                        4,
                         "clean, high-resolution, 8k, best quality, masterpiece",
                         "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
                         50,
@@ -338,11 +347,32 @@ with block:
                         320,
                         4,
                         1,
-                    ], 
+                    ],
+                    [
+                        "preset/datasets/test_datasets/man.png", 
+                        "",
+                        True,
+                        0.1,
+                        4,
+                        4,
+                        "clean, high-resolution, 8k, best quality, masterpiece",
+                        "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
+                        50,
+                        4,
+                        7.5,
+                        123,
+                        320,
+                        4,
+                        1,
+                    ],                    
                 ],
                 inputs=[
                     input_image,
                     user_prompt,
+                    use_KDS,
+                    bandwidth,
+                    patch_size,
+                    num_particles,
                     positive_prompt,
                     negative_prompt,
                     num_inference_steps,
@@ -360,6 +390,10 @@ with block:
     inputs = [
         input_image,
         user_prompt,
+        use_KDS,
+        bandwidth,
+        patch_size,
+        num_particles,
         positive_prompt,
         negative_prompt,
         num_inference_steps,
@@ -373,4 +407,3 @@ with block:
     run_button.click(fn=process, inputs=inputs, outputs=[result_gallery])
 
 block.launch(share=True)
-