Update app.py

app.py

@@ -6,7 +6,7 @@ from PIL import Image, ImageOps
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipeline = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to("cuda")
 
-def generate(image_editor, prompt):
+def generate(image_editor, prompt, neg_prompt, strength, guidance):
     image = image_editor['background'].convert('RGB')
     mask = Image.new("RGBA", image_editor["layers"][0].size, "WHITE") 
     mask.paste(image_editor["layers"][0], (0, 0), image_editor["layers"][0])
@@ -15,7 +15,14 @@ def generate(image_editor, prompt):
     image.thumbnail((1024, 1024))
     mask.thumbnail((1024, 1024))
 
-    final_image = pipeline(prompt=prompt, image=image, mask_image=mask).images[0]
+    final_image = pipeline(prompt=prompt, 
+                           image=image, 
+                           mask_image=mask, 
+                           width=image.width, 
+                           height=image.height, 
+                           num_inference_steps=50, 
+                           strength=strength, 
+                           guidance_scale=guidance).images[0]
 
     return image_editor, image, mask, final_image
 
@@ -44,6 +51,6 @@ with gr.Blocks() as demo:
         out2 = gr.Image()
         out3 = gr.Image()
         
-    generate_button.click(fn=generate, inputs=[sketch_pad,prompt], outputs=[sketch_pad, out1, out2, out3])
+    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, strength_slider, guidance_slider], outputs=[sketch_pad, out1, out2, out3])
 
 demo.launch()