Update app.py

app.py

@@ -18,11 +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)
+    alpha_channel = (mask*255).astype(np.uint8)
+    print(fg.max(), alpha_channel.max(), fg.shape, alpha_channel.shape)
     print("***********Inference finish****************")
     # print("***********MASK****************", inp_np.max(), mask.max())
-    fg = np.dstack((fg, mask))
+    fg = np.dstack((fg, alpha_channel))
     fg_pil = Image.fromarray(fg, 'RGBA')
     
     return [mask, fg_pil]