incendies Cursor commited on
Commit
f9d0bb5
·
1 Parent(s): 392b2d6

Fix bg remover: original_image=None, use hf-applications/background-removal

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,14 +1,15 @@
1
  import gradio as gr
2
  from daggr import GradioNode, Graph
3
 
4
- # Step 1: Background Removal (merve/background-removal returns origin, processed_image)
 
5
  bg_remover = GradioNode(
6
- "merve/background-removal",
7
  api_name="/image",
8
  run_locally=False,
9
  inputs={"image": gr.Image()},
10
  outputs={
11
- "origin": gr.Image(label="Original"),
12
  "final_image": gr.Image(label="Processed Image"),
13
  },
14
  )
 
1
  import gradio as gr
2
  from daggr import GradioNode, Graph
3
 
4
+ # Background removal: API returns (original, processed). Hide first with None, show final_image.
5
+ # hf-applications/background-removal runs on GPU (Zero); use merve/background-removal for CPU.
6
  bg_remover = GradioNode(
7
+ "hf-applications/background-removal",
8
  api_name="/image",
9
  run_locally=False,
10
  inputs={"image": gr.Image()},
11
  outputs={
12
+ "original_image": None,
13
  "final_image": gr.Image(label="Processed Image"),
14
  },
15
  )