Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -477,7 +477,7 @@ class DemoUI(object):
|
|
477 |
et = time.time()
|
478 |
msg = f"prompt: {prompt}; seed: {seed}; cost time: {et - st}s; repaiting scale: {repainting_scale}"
|
479 |
if pre_edit_image is not None:
|
480 |
-
ret_image = Image.composite(
|
481 |
else:
|
482 |
ret_image = None
|
483 |
return (gr.Image(value=image), gr.Column(visible=True),
|
|
|
477 |
et = time.time()
|
478 |
msg = f"prompt: {prompt}; seed: {seed}; cost time: {et - st}s; repaiting scale: {repainting_scale}"
|
479 |
if pre_edit_image is not None:
|
480 |
+
ret_image = Image.composite(Image.new("RGB", pre_edit_image.size, (0, 0, 0)), pre_edit_image, pre_edit_mask)
|
481 |
else:
|
482 |
ret_image = None
|
483 |
return (gr.Image(value=image), gr.Column(visible=True),
|