Update app.py

app.py

@@ -3,18 +3,29 @@ 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
+    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
 
+# Custom CSS to hide Gradio branding and adjust margins
 app = gr.Interface(
-    fn= remove_background, #função
-    inputs= gr.components.Image(type='pil'), #entrada (anexo do usuário)
-    outputs= gr.components.Image(type='pil', format='png'), #saida (download da imagem sem fundo no formato PNG)
-    title= 'Remove image background',
-    description= 'Remove backgrounds from photos automatically in seconds - 100% free, no watermarks, works on any device!'
+    fn=remove_background,
+    inputs=gr.components.Image(type='pil'),
+    outputs=gr.components.Image(type='pil', format='png'),
+    title='Remove image background',
+    description='Remove backgrounds from photos automatically in seconds - 100% free, no watermarks, works on any device!'
 )
 
+# Remove Gradio footer and API options
+app.css = """
+footer {display: none !important;}
+.gradio-container {margin-bottom: 0 !important;}
+"""
+
 if __name__ == "__main__":
-    app.launch(share=True)  # Disables the API option
+    app.launch(
+        share=True,      # Optional: Remove if you don't need sharing
+        show_api=False, # Hides the "Use via API" option
+        show_tips=False # Disables Gradio tips for cleaner UI
+    )