rishi9440 commited on
Commit
4aa7b1c
·
1 Parent(s): 441db3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  from rembg import remove
3
  from PIL import Image
4
- import os
5
 
6
  def remove_background(input_image):
7
 
@@ -18,7 +17,7 @@ iface = gr.Interface(remove_background,
18
  outputs=output_image,
19
  title='Image Background Remover',
20
  description='Upload an image and it will remove.',
21
- os.path.join(os.path.dirname(__file__), "images/lion.jpg")
22
  )
23
 
24
  # Launch the interface
 
1
  import gradio as gr
2
  from rembg import remove
3
  from PIL import Image
 
4
 
5
  def remove_background(input_image):
6
 
 
17
  outputs=output_image,
18
  title='Image Background Remover',
19
  description='Upload an image and it will remove.',
20
+ Image.save(output_image)
21
  )
22
 
23
  # Launch the interface