Update app.py

app.py

@@ -5,12 +5,12 @@ import subprocess
 import os
 import datetime
 from PIL import Image
-# from remove_bg import RemoveBackground
+from remove_bg import RemoveBackground
 
 net_C = hf_hub_download("radames/PIFu-upright-standing", filename="net_C")
 net_G = hf_hub_download("radames/PIFu-upright-standing", filename="net_G")
 
-# remove_bg = RemoveBackground()
+remove_bg = RemoveBackground()
 
 env = {
     **os.environ,
@@ -24,15 +24,11 @@ def process(img_path):
     base = os.path.basename(img_path)
     img_name = os.path.splitext(base)[0]
     print("image name", img_name)
-    #img = Image.open(img_path)
+    img = Image.open(img_path)
     # remove background
-    #print("remove background")
-    #foreground = Image.fromarray(remove_bg.inference(img), 'RGBA')
-    #foreground.save("./PIFu/inputs/" + img_name + ".png")
     print("remove background")
-    subprocess.Popen(["backgroundremover", "-i", img_path, "-o", f'./PIFu/inputs/{img_name}.png']).communicate()
-    print("remove background Done")
-    
+    foreground = Image.fromarray(remove_bg.inference(img), 'RGBA')
+    foreground.save("./PIFu/inputs/" + img_name + ".png")
     print("align mask with input training image")
     subprocess.Popen(["python", "./apps/crop_img.py", "--input_image",
                       f'./inputs/{img_name}.png', "--out_path", "./inputs"], cwd="PIFu").communicate()