Update app.py

app.py

@@ -18,9 +18,11 @@ def predict(inp):
     mask = inference_img(model, inp) 
     inp_np = np.array(inp)
     fg = np.uint8((mask[:,:,None]*inp_np))
+    mask = (mask > 0.5).astype(np.uint8)
+    print(fg.max(), mask.max(), fg.shape, mask.shape)
     print("***********Inference finish****************")
     # print("***********MASK****************", inp_np.max(), mask.max())
-    # fg = np.dstack((inp_np, mask))
+    fg = np.dstack((fg, mask))
     fg_pil = Image.fromarray(fg, 'RGBA')
     
     return [mask, fg_pil]