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