Upload 3 files

README.md

@@ -0,0 +1,14 @@
+---
+title: RMBG
+emoji: 🪄
+colorFrom: green
+colorTo: yellow
+sdk: gradio
+sdk_version: 5.9.1
+app_file: app.py
+pinned: false
+license: creativeml-openrail-m
+header: mini
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,37 @@
+import gradio as gr
+from rembg import remove
+
+# Custom CSS for styling
+css = '''
+.gradio-container { max-width: 1200px !important; }
+h1 { text-align: center; }
+footer { visibility: hidden; }
+'''
+
+# Function to remove background
+def segment(image):
+    """
+    Removes the background from the input image.
+
+    Args:
+        image: An image file uploaded by the user.
+
+    Returns:
+        The image with its background removed.
+    """
+    return remove(image)
+
+# Gradio Interface
+demo = gr.Interface(
+    fn=segment,
+    inputs=gr.Image(label="Input Image", interactive=True),
+    outputs=gr.Image(label="Result Image"),
+    title="RMBG",
+    description="Upload an image to remove its background automatically!",
+    css=css,
+    theme="bethecloud/storj_theme"  # Ensure this theme is available or remove it
+)
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch(show_api=False)

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+rembg
+onnxruntime
+pillow