Spaces:
Runtime error
Runtime error
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()
|