Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ def draw_on_image(image, prompt):
|
|
29 |
image["mask"]
|
30 |
)
|
31 |
|
32 |
-
res_image = pipeline(prompt=prompt, image=init_image, mask_image=mask_image).images[0]
|
33 |
|
34 |
return res_image
|
35 |
|
|
|
29 |
image["mask"]
|
30 |
)
|
31 |
|
32 |
+
res_image = pipeline(prompt=prompt, image=init_image, mask_image=mask_image, num_inference_steps=2).images[0]
|
33 |
|
34 |
return res_image
|
35 |
|