Update app.py

app.py

@@ -134,6 +134,16 @@ def inference(image: Image):
   im_rgba = im_rgb.copy()
   im_rgba.putalpha(pil_mask)
 
+  # Apply Background
+  bg_color = 0x00C49A
+  w = img_rgba.size[0]
+  h = img_rgba.size[1]
+    
+  bg = Image.new("RGB", (w,h), bg_color)
+  im_colored = Image.new("RGBA", (w,h))
+  im_colored.paste(bg)
+  im_colored.paste(im_rgba)
+
   return [im_rgba, pil_mask]