Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def predict(source_img, prompt, negative_prompt):
|
|
22 |
src.save("src.png")
|
23 |
mask = resize(512, "data_mask.png")
|
24 |
mask.save("mask.png")
|
25 |
-
image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=src, mask_image=mask, num_inference_steps=
|
26 |
return image
|
27 |
|
28 |
title="SDXL 1.0 Inpainting CPU"
|
|
|
22 |
src.save("src.png")
|
23 |
mask = resize(512, "data_mask.png")
|
24 |
mask.save("mask.png")
|
25 |
+
image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=src, mask_image=mask, num_inference_steps=5).images[0]
|
26 |
return image
|
27 |
|
28 |
title="SDXL 1.0 Inpainting CPU"
|