Update app.py

app.py

@@ -291,7 +291,7 @@ def edit_inference(net, prompt, negative_prompt, guidance_scale, ddim_steps, see
     image = image.detach().cpu().float().permute(0, 2, 3, 1).numpy()[0] 
     image = Image.fromarray((image * 255).round().astype("uint8"))
 
-    return net, (image, input_image["background"])
+    return net, image
 
 
 class CustomImageDataset(Dataset):