Update app.py

app.py

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