Update app.py

app.py

@@ -128,7 +128,7 @@ def inference(image: Image):
   image_tensor, orig_size = load_image(image_path, hypar) 
   mask = predict(net, image_tensor, orig_size, hypar, device)
   
-  pil_mask = Image.fromarray(mask).convert('1')
+  pil_mask = Image.fromarray(mask).convert("1").convert("L")
   im_rgb = Image.open(image).convert("RGB")
   
   im_rgba = im_rgb.copy()