Spaces:

befozg
/

stylematte

Running

befozg commited on Sep 8, 2023

Commit

e5113ec

•

1 Parent(s): 3139aef

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ def predict(inp):
     fg = np.uint8((mask[:,:,None]*inp_np))
     print("***********Inference finish****************")
     # print("***********MASK****************", inp_np.max(), mask.max())
-    fg_pil = Image.fromarray(fg, 'RGBA')
     return [mask, fg_pil]

     fg = np.uint8((mask[:,:,None]*inp_np))
     print("***********Inference finish****************")
     # print("***********MASK****************", inp_np.max(), mask.max())
+    fg_pil = Image.fromarray(fg)
     return [mask, fg_pil]