Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -128,7 +128,7 @@ def inference(image: Image):
|
|
| 128 |
image_tensor, orig_size = load_image(image_path, hypar)
|
| 129 |
mask = predict(net, image_tensor, orig_size, hypar, device)
|
| 130 |
|
| 131 |
-
pil_mask = Image.fromarray(mask).convert("
|
| 132 |
im_rgb = Image.open(image).convert("RGB")
|
| 133 |
|
| 134 |
im_rgba = im_rgb.copy()
|
|
|
|
| 128 |
image_tensor, orig_size = load_image(image_path, hypar)
|
| 129 |
mask = predict(net, image_tensor, orig_size, hypar, device)
|
| 130 |
|
| 131 |
+
pil_mask = Image.fromarray(mask).convert("L")
|
| 132 |
im_rgb = Image.open(image).convert("RGB")
|
| 133 |
|
| 134 |
im_rgba = im_rgb.copy()
|