Update app.py

app.py

@@ -53,8 +53,6 @@ def sample(zs, wts, prompt_tar="", skip=36, cfg_scale_tar=15, eta = 1):
 
 # load pipelines
 sd_model_id = "runwayml/stable-diffusion-v1-5"
-# sd_model_id = "CompVis/stable-diffusion-v1-4"
-# sd_model_id = "stabilityai/stable-diffusion-2-base"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
 sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")
@@ -156,11 +154,6 @@ with gr.Blocks(css='style.css') as demo:
             seed = 0,
             randomize_seed  = True):
 
-        # if randomized_seed:
-        #     seed = random.randint(0, np.iinfo(np.int32).max)
-            
-        # torch.manual_seed(seed)
-        #  # offsets=(0,0,0,0)
         x0 = load_512(input_image, device=device)
     
         if do_inversion or randomize_seed:
@@ -173,7 +166,6 @@ with gr.Blocks(css='style.css') as demo:
         return output, wts, zs, do_inversion
     
     gr.HTML(intro)
-    # xt = gr.State(value=False)
     wts = gr.State()
     zs = gr.State()
     do_inversion = gr.State(value=True)
@@ -210,7 +202,7 @@ with gr.Blocks(css='style.css') as demo:
     edit_button.click(
         fn = randomize_seed_fn,
         inputs = [seed, randomize_seed],
-        outputs = [seed]).then(
+        outputs = [seed], queue = False).then(
         fn=edit,
         inputs=[input_image,
                 do_inversion, wts, zs, 
@@ -235,10 +227,6 @@ with gr.Blocks(css='style.css') as demo:
         outputs = [do_inversion]
     )
 
-    # skip.change(
-    #     fn = reset_latents
-    # )
-
 
     gr.Examples(
         label='Examples',