Update app.py

app.py

@@ -3,16 +3,15 @@ from rembg import remove
 from PIL import Image
 
 def remove_bg(image):
-    output_image = remove(image)
-    return output_image
+    return remove(image)
 
-# Create Gradio interface
-demo = gr.Interface(
-    fn=remove_bg,
-    inputs=gr.Image(type="pil"),  # Accept PIL images
-    outputs="image",  # Output as an image
-)
+with gr.Blocks() as demo:
+    gr.Markdown("## U²Net Background Remover API")
+    with gr.Row():
+        inp = gr.Image(type="pil")
+        out = gr.Image()
+    submit = gr.Button("Remove Background")
+    submit.click(remove_bg, inputs=inp, outputs=out)
 
-# Launch the app
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()