Update app.py
Browse files
app.py
CHANGED
|
@@ -228,14 +228,14 @@ model_inpaint = None
|
|
| 228 |
def inpaint(image, invert):
|
| 229 |
# image = resize_image(img, res)
|
| 230 |
# color = HWC3(image["image"])
|
| 231 |
-
color = HWC3(image["backbround"])
|
| 232 |
if(invert):
|
| 233 |
# alpha = image["mask"][:, :, 0:1]
|
| 234 |
alpha = image["layers"][0][:, :, 0:1]
|
| 235 |
else:
|
| 236 |
# alpha = 255 - image["mask"][:, :, 0:1]
|
| 237 |
alpha = 255 - image["layers"][0][:, :, 0:1]
|
| 238 |
-
result = np.concatenate([
|
| 239 |
return [result]
|
| 240 |
|
| 241 |
|
|
|
|
| 228 |
def inpaint(image, invert):
|
| 229 |
# image = resize_image(img, res)
|
| 230 |
# color = HWC3(image["image"])
|
| 231 |
+
# color = HWC3(image["backbround"])
|
| 232 |
if(invert):
|
| 233 |
# alpha = image["mask"][:, :, 0:1]
|
| 234 |
alpha = image["layers"][0][:, :, 0:1]
|
| 235 |
else:
|
| 236 |
# alpha = 255 - image["mask"][:, :, 0:1]
|
| 237 |
alpha = 255 - image["layers"][0][:, :, 0:1]
|
| 238 |
+
result = np.concatenate([image["backbround"], alpha], axis=2)
|
| 239 |
return [result]
|
| 240 |
|
| 241 |
|