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