Spaces:
Build error
Build error
Update app.py
Browse filesRemoved Strength
app.py
CHANGED
@@ -24,7 +24,7 @@ def predict(source_img, prompt, negative_prompt):
|
|
24 |
src.save("src.png")
|
25 |
mask = resize(512, "data_mask.png")
|
26 |
mask.save("mask.png")
|
27 |
-
image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=src, mask_image=mask,
|
28 |
return image
|
29 |
|
30 |
title="Stable Diffusion 2.0 Inpainting CPU"
|
|
|
24 |
src.save("src.png")
|
25 |
mask = resize(512, "data_mask.png")
|
26 |
mask.save("mask.png")
|
27 |
+
image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=src, mask_image=mask, num_inference_steps=10).images[0]
|
28 |
return image
|
29 |
|
30 |
title="Stable Diffusion 2.0 Inpainting CPU"
|