Update app.py

app.py

@@ -1,30 +1,30 @@
-import gradio as gr
-from transformers import pipeline
-from PIL import Image
-
-# Função para remover o background da imagem
-def remove_background(image):
-    pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True)
-    # Obter a máscara da imagem
-    segmentation = pipe(image)
-    # Aplicar a máscara para remover o fundo
-    # Filtrando a parte da segmentação onde a label é "person" ou qualquer outra desejada
-    mask = Image.fromarray(segmentation[0]['mask'])  # Segmentação com maior probabilidade
-    original_image = Image.fromarray(image)
-    
-    # Remover fundo
-    output = Image.composite(original_image, Image.new("RGBA", original_image.size), mask)
-    return output
-
-# Criar a interface Gradio
-app = gr.Interface(
-    fn=remove_background,
-    inputs=gr.Image(type="pil"),
-    outputs=gr.Image(type="pil", format="png"),  # Especificar a 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."
-)
-
-# Executar a aplicação
-if __name__ == "__main__":
-    app.launch(share=True)
+import gradio as gr
+from transformers import pipeline
+from PIL import Image
+
+# Função para remover o background da imagem
+def remove_background(image):
+    pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True)
+    # Obter a máscara da imagem
+    segmentation = pipe(image)
+    # Aplicar a máscara para remover o fundo
+    # Filtrando a parte da segmentação onde a label é "person" ou qualquer outra desejada
+    mask = Image.fromarray(segmentation[0]['mask'])  # Segmentação com maior probabilidade
+    original_image = Image.fromarray(image)
+    
+    # Remover fundo
+    output = Image.composite(original_image, Image.new("RGBA", original_image.size), mask)
+    return output
+
+# Criar a interface Gradio
+app = gr.Interface(
+    fn=remove_background,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
+    title="Remoção de Background de Imagens",
+    description="Envie uma imagem e veja o background sendo removido automaticamente."
+)
+
+# Executar a aplicação
+if __name__ == "__main__":
+    app.launch(share=True)