Update main.py

main.py

@@ -0,0 +1,12 @@
+import gradio as gr
+from rembg import remove
+from PIL import Image
+
+def remove_background(input_image):
+    output_image = remove(input_image)
+    return output_image
+
+inputs = gr.inputs.Image()
+outputs = gr.outputs.Image(type='pil')
+
+gr.Interface(fn=remove_background, inputs=inputs, outputs=outputs).launch(share=True)