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):
|