Update example_inference.py

example_inference.py

@@ -28,10 +28,10 @@ def example_inference():
     result_image = postprocess_image(result[0][0], orig_im_size)
 
     # save result
-    pil_im = Image.fromarray(result_image)
-    no_bg_image = Image.new("RGBA", pil_im.size, (0,0,0,0))
+    pil_mask_im = Image.fromarray(result_image)
     orig_image = Image.open(im_path)
-    no_bg_image.paste(orig_image, mask=pil_im)
+    no_bg_image = orig_image.copy()
+    no_bg_image.putalpha(pil_mask_im)    
     no_bg_image.save("example_image_no_bg.png")