Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def remove_obj(image, seed):
|
|
35 |
alpha_channel = image["layers"][0][:, :, 3]
|
36 |
mask = np.where(alpha_channel == 0, 0, 255).astype(np.uint8)
|
37 |
uploaded_mask = Image.fromarray(mask)
|
38 |
-
background = Image.fromarray(
|
39 |
|
40 |
mask = dilate_mask(uploaded_mask)
|
41 |
seed = int(seed)
|
|
|
35 |
alpha_channel = image["layers"][0][:, :, 3]
|
36 |
mask = np.where(alpha_channel == 0, 0, 255).astype(np.uint8)
|
37 |
uploaded_mask = Image.fromarray(mask)
|
38 |
+
background = Image.fromarray(image["background"])
|
39 |
|
40 |
mask = dilate_mask(uploaded_mask)
|
41 |
seed = int(seed)
|