updates

app.py

@@ -1,7 +1,5 @@
 import PIL.Image
 import gradio as gr
-import huggingface_hub
-import onnxruntime as rt
 import numpy as np
 import cv2
 from PIL import ImageOps
@@ -36,11 +34,16 @@ def get_mask(img_in):
         result = response.json()
         print('Result:', result)
 
+        bg_img = Image.open(
+            "/examples/bg1.jpg").convert("RGBA").resize(img_in.size)
+
         mask = Image.open(io.BytesIO(base64.b64decode(
             result['mask']))).resize(img_in.size)
         img_in = img_in.convert("RGBA")
         img_in.putalpha(mask)
 
+        img_in = Image.alpha_composite(bg_img, img_in)
+
         return (img_in, result['emotion'])
     else:
         print('error:', response.text)