Spaces:
Running
Running
File size: 1,319 Bytes
5658f28 208c40c 5f5d761 4d8d03d da190f1 208c40c 5f5d761 5658f28 5f5d761 5658f28 5f5d761 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
from rembg import remove
from PIL import Image, ImageOps
import gradio as gr
def change_color(passport='Passport Photo', color='Background Color', background='Photo Background'):
if background:
image = remove(passport)
else:
image = passport
new_image = Image.new("RGB", image.size, color)
new_image.paste(image, mask=image)
# Add a black border of 10 pixels on all sides
new_image = ImageOps.expand(new_image, border=5, fill='black')
new_image = ImageOps.expand(new_image, border=1, fill='white')
# Get the image size
width, height = new_image.size
# Create a new image to hold the photos
output_image = Image.new("RGB", (width * 4, height * 2))
# Paste the source image into the new image
for row in range(2):
for col in range(4):
output_image.paste(new_image, (col * width, row * height))
# Save the output image
return output_image
inputs = [
gr.Image(label="Passport Photo", type="pil", image_mode="RGBA"),
gr.ColorPicker(label="Background Color",value="#ff00ff"),
gr.Checkbox(label="Select if you want to remove background", value=True)
]
outputs = gr.Image(label="Download Image")
demo = gr.Interface(
fn=change_color,
inputs=inputs,
outputs=outputs,
)
if __name__ == "__main__":
demo.launch() |