Upload app.py
Browse files
app.py
CHANGED
@@ -217,7 +217,7 @@ model_inpaint = None
|
|
217 |
def inpaint(image):
|
218 |
# image = resize_image(img, res)
|
219 |
color = HWC3(image["image"])
|
220 |
-
alpha =
|
221 |
result = np.concatenate([color, alpha], axis=2)
|
222 |
return [result]
|
223 |
|
|
|
217 |
def inpaint(image):
|
218 |
# image = resize_image(img, res)
|
219 |
color = HWC3(image["image"])
|
220 |
+
alpha = image["mask"][:, :, 1]
|
221 |
result = np.concatenate([color, alpha], axis=2)
|
222 |
return [result]
|
223 |
|