Create app.py

app.py

@@ -0,0 +1,31 @@
+import gradio as gr
+import rembg
+from PIL import Image
+import io
+import numpy as np
+
+def remove_background(input_image):
+ # Convert Gradio input image data to bytes
+ image_array = input_image.astype(np.uint8) # Ensure the image is in the correct format
+ image_pil = Image.fromarray(image_array)
+
+ # Save the PIL Image to an in-memory buffer
+ image_buffer = io.BytesIO()
+ image_pil.save(image_buffer, format='PNG')
+
+ # Use rembg to remove the background
+ output_data = rembg.remove(image_buffer.getvalue())
+
+ # Convert the output data to a PIL Image
+ output_image = Image.open(io.BytesIO(output_data))
+
+ # Return the output image directly
+ return np.array(output_image)
+
+iface = gr.Interface(
+ fn=remove_background,
+ inputs=gr.Image(),
+ outputs=gr.Image()
+)
+
+iface.launch()