Aryan Wadhawan commited on
Commit
96da21c
1 Parent(s): 73f00ae
Files changed (2) hide show
  1. __pycache__/u2net.cpython-312.pyc +0 -0
  2. app.py +1 -1
__pycache__/u2net.cpython-312.pyc CHANGED
Binary files a/__pycache__/u2net.cpython-312.pyc and b/__pycache__/u2net.cpython-312.pyc differ
 
app.py CHANGED
@@ -89,7 +89,7 @@ def predict(image):
89
  iface = gr.Interface(
90
  fn=predict,
91
  inputs=gr.Image(type="pil"),
92
- outputs=gr.Image(type="pil", label="Edited Image"), # Adjust the box size
93
  title="Background Removal with U2NET",
94
  description="Upload an image and remove the background"
95
  )
 
89
  iface = gr.Interface(
90
  fn=predict,
91
  inputs=gr.Image(type="pil"),
92
+ outputs=gr.Image(type="pil", label="Edited Image", image_mode="RGBA"), # RGBA ensures PNG with transparency
93
  title="Background Removal with U2NET",
94
  description="Upload an image and remove the background"
95
  )