Create app.py

app.py

@@ -0,0 +1,39 @@
+import gradio as gr
+from rembg import remove
+from PIL import Image
+import io
+
+def background_remover(input_image):
+    input_bytes = input_image.read()  
+    input_img = Image.open(io.BytesIO(input_bytes))
+    output_img = remove(input_img) 
+    
+    buf = io.BytesIO()
+    output_img.save(buf, format='PNG')
+    buf.seek(0)
+    return buf
+
+description = """
+This app uses the rembg library to remove the background from uploaded images.
+Simply upload your image and let the model do its work. Download the result immediately after!
+"""
+
+with gr.Blocks() as demo:
+    gr.Markdown("<h1 style='text-align: center;'>Background Remover</h1>")
+    gr.Markdown(description)
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            input_image = gr.Image(type='file', label="Upload Image")
+        with gr.Column(scale=1):
+            output_image = gr.Image(type='pil', label="Result")
+
+    with gr.Row():
+         gr.Button("Remove Background").click(
+             fn=background_remover,
+             inputs=input_image,
+             outputs=output_image
+         )
+
+    demo_examples = [["woman.jpg"], ["groot.jpg"]]
+    demo.launch(share=True, examples=demo_examples, theme="soft")