Spaces:
Runtime error
Runtime error
PyWebSolutions
commited on
Commit
·
4e4eddd
1
Parent(s):
9e4ac2b
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, num_inference_steps=
|
| 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=5).images[0]
|
| 33 |
|
| 34 |
return res_image
|
| 35 |
|