Update app.py

app.py

@@ -123,7 +123,7 @@ hypar["model"] = ISNetDIS()
 net = build_model(hypar, device)
 
 
-def inference(image: Image):
+def inference(image: Image, color: str):
   image_path = image
   
   image_tensor, orig_size = load_image(image_path, hypar) 
@@ -151,8 +151,14 @@ article = "<div><center><img src='https://visitor-badge.glitch.me/badge?page_id=
 
 interface = gr.Interface(
     fn=inference,
-    inputs=gr.Image(type='filepath'),
-    outputs=["image", "image"],
+    inputs=[
+        gr.Image(type='filepath', label="Image"),
+        gr.ColorPicker(label="Background Color", value="0x00C49A")
+    ],
+    outputs=[
+        gr.Image(label="Full Background"),
+        gr.Image(label="Transparent Background")
+    ],
     examples=[['robot.png'], ['ship.png']],
     title=title,
     description=description,