File size: 680 Bytes
bb13d4b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2d1c6d7
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import gradio as gr
from rembg import remove
from PIL import Image

def remove_background(input_image):

    input = Image.open(input_image) # load image
    output = remove(input) # remove background
    return output


input_image = gr.inputs.Image(type="filepath", label="Input Image")
output_image = gr.inputs.Image(type="filepath", label="Output Image")
# Create a Gradio interface
iface = gr.Interface(remove_background, 
                     inputs=input_image, 
                     outputs=output_image, 
                     title='Image Background Remover',
                     description='Upload an image and it will remove.')

# Launch the interface
iface.launch()