Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| from PIL import Image | |
| def remove_background(image): | |
| pipe = pipeline("image-segmentation", model = "briaai/RMBG-1.4", trust_remote_code = True) | |
| pillow_mask = pipe(image, return_mask = True) | |
| pillow_image = pipe(image) | |
| return pillow_image | |
| app = gr.Interface( | |
| fn = remove_background, | |
| inputs = gr.components.Image(type = "pil"), | |
| outputs = gr.components.Image(type = "pil", format = "png"), | |
| title = "Remoção de Fundo de Imagens", | |
| description = "Utilize o modelo para enviar uma imagem e receba a imagem sem o fundo." | |
| ) | |
| if __name__ == "__main__": | |
| app.launch(share = True) |