Update app.py

app.py

@@ -30,7 +30,7 @@ model.load_state_dict(load_state_dict(ckpt), strict=False)
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)
 
-
+[email protected]
 def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_steps, guess_mode, strength, scale, seed, eta, low_threshold, high_threshold):
     with torch.no_grad():
         img = resize_image(HWC3(input_image), image_resolution)
@@ -74,7 +74,7 @@ def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resoluti
         x_samples = (einops.rearrange(x_samples, 'b c h w -> b h w c') * 127.5 + 127.5).cpu().numpy().clip(0, 255).astype(np.uint8)
 
         results = [x_samples[i] for i in range(num_samples)]
-    return [255 - img] + results
+    return results
 
 
 block = gr.Blocks().queue()