Update app.py

app.py

@@ -22,7 +22,7 @@ def resize_image(image, model_input_size=(1024, 1024)):
     image = image.resize(model_input_size, Image.BILINEAR)
     return image
 
-def process(image):
+def process(image, background_image=None):
     # prepare input
     orig_image = Image.fromarray(image).convert("RGBA")
     w, h = orig_im_size = orig_image.size
@@ -47,7 +47,7 @@ def process(image):
     final_image = Image.new("RGBA", orig_image.size)
     final_image.paste(orig_image, mask=mask_image)
 
-    return final_image
+    return new_im  # or any other variable you wish to return
 
 
 def merge_images(background_image, foreground_image):
@@ -74,17 +74,13 @@ def merge_images(background_image, foreground_image):
     return background
 
 
-# Gradio 인터페이스 정의
 demo = gr.Interface(
     fn=process,
     inputs=[
-        gr.Image(type="numpy", label="Image to remove background"),
-        gr.Image(type="pil", label="Optional: Background Image")
+        gr.Image(type="numpy", label="Image to remove background"),  # 첫 번째 인자
+        gr.Image(type="pil", label="Optional: Background Image", optional=True)  # 두 번째 인자, 선택적
     ],
     outputs="image",
-    title="Background Removal",
-    description="This is a demo for BRIA RMBG 1.4 that uses the BRIA RMBG-1.4 image matting model as backbone."
+    title=title,
+    description=description
 )
-
-if __name__ == "__main__":
-    demo.launch()