Update app.py

app.py

@@ -15,22 +15,6 @@ from gradio_imageslider import ImageSlider
 
 print(f'torch version:{torch.__version__}')
 
-# import subprocess
-# import importlib, site, sys
-
-# # Re-discover all .pth/.egg-link files
-# for sitedir in site.getsitepackages():
-#     site.addsitedir(sitedir)
-
-# # Clear caches so importlib will pick up new modules
-# importlib.invalidate_caches()
-
-# def sh(cmd): subprocess.check_call(cmd, shell=True)
-
-# sh("pip install -U xformers --index-url https://download.pytorch.org/whl/cu126")
-
-# # tell Python to re-scan site-packages now that the egg-link exists
-# import importlib, site; site.addsitedir(site.getsitepackages()[0]); importlib.invalidate_caches()
 
 import torch.utils.checkpoint
 from pytorch_lightning import seed_everything
@@ -192,10 +176,6 @@ 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,
@@ -254,7 +234,6 @@ def process(
                     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, bandwidth=bandwidth, num_particles=num_particles, patch_size=patch_size,
                 ).images[0]
 
             if True:  # alpha<1.0:
@@ -315,61 +294,6 @@ with block:
                     [
                         "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,
-                        4,
-                        7.5,
-                        123,
-                        320,
-                        4,
-                        1,
-                    ],
-                    [
-                        "preset/datasets/test_datasets/man.png", 
-                        "",
-                        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,
-                        4,
-                        7.5,
-                        123,
-                        320,
-                        4,
-                        1,
-                    ],
-                    [
-                        "preset/datasets/test_datasets/man.png", 
-                        "",
-                        True,
-                        0.1,
-                        4,
-                        16,
-                        "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,
-                    ],     
-                    [
-                        "preset/datasets/test_datasets/man.png", 
-                        "",
-                        True,
-                        0.1,
-                        16,
-                        16,
                         "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,
@@ -379,15 +303,11 @@ with block:
                         320,
                         4,
                         1,
-                    ],                    
+                    ],               
                 ],
                 inputs=[
                     input_image,
                     user_prompt,
-                    use_KDS,
-                    bandwidth,
-                    patch_size,
-                    num_particles,
                     positive_prompt,
                     negative_prompt,
                     num_inference_steps,
@@ -405,10 +325,6 @@ with block:
     inputs = [
         input_image,
         user_prompt,
-        use_KDS,
-        bandwidth,
-        patch_size,
-        num_particles,
         positive_prompt,
         negative_prompt,
         num_inference_steps,