1
resposta

[Faça como eu fiz]: criando a aplicação e adicionando ao Spaces

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

gradio_app = gr.Interface(
fn=remove_background,
inputs=gr.components.Image(type="pil"),
outputs=gr.components.Image(type="pil", format="png"), # Especificar saída como PNG
title="Remoção de Background de Imagens",
description="Envie uma imagem e veja o background sendo removido automaticamente. A imagem resultante será no formato PNG."
)

if name == "main":
gradio_app.launch(share=True)

Garanta sua matrícula hoje e ganhe + 2 meses grátis

Continue sua jornada tech com ainda mais tempo para aprender e evoluir

Quero aproveitar agora
1 resposta

Oi, Luan! Como vai?

Agradeço por compartilhar seu código com a comunidade Alura.

Achei muito legal sua iniciativa de criar uma interface no Gradio para remover o background das imagens. A forma como você estruturou a função remove_background mostra bastante clareza na organização do código.

Uma dica interessante para o futuro é usar o método type() para verificar o tipo do objeto que está sendo manipulado. Isso pode ajudar a evitar erros em funções que recebem diferentes entradas. Veja este exemplo:


valor = 10
print(type(valor))  # Mostra que valor é um inteiro

Esse código imprime o tipo da variável, ajudando a entender melhor o que está sendo processado.

Alura

Conte com o apoio da comunidade Alura na sua jornada. Abraços e bons estudos!