Update app.py

app.py

@@ -18,16 +18,17 @@ def predict(inp):
     inp_np = np.array(inp)
     fg = np.uint8((mask[:,:,None]*inp_np))
     print("***********Inference finish****************")
-    print("***********MASK****************", inp_np.max(), mask.max())
+    # print("***********MASK****************", inp_np.max(), mask.max())
+    fg_pil = Image.fromarray(fg, 'RGBA')
     
-    return [mask, fg]
+    return [mask, fg_pil]
 
 print("MODEL LOADED")
 print("************************************")
 
 iface = gr.Interface(fn=predict, 
              inputs=gr.Image(type="numpy"),
-             outputs=[gr.Image(type="numpy"),gr.Image(type="numpy", image_mode='RGBA')],
+             outputs=[gr.Image(type="numpy"),gr.Image(type="pil", image_mode='RGBA')],
              examples=["./logo.jpeg"])
 print("****************Interface created******************")