File size: 666 Bytes
cf5c0c0 2c9a089 1138828 2c9a089 4b505e1 2c9a089 1138828 2c9a089 6dcc89e cf5c0c0 2c9a089 cf5c0c0 |
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 |
import spaces
import gradio as gr
from transparent_background import Remover
from PIL import Image
import numpy as np
@spaces.GPU(duration=20)
def remove_background(image):
remover = Remover()
if isinstance(image, Image.Image):
output = remover.process(image)
elif isinstance(image, np.ndarray):
image_pil = Image.fromarray(image)
output = remover.process(image_pil)
else:
raise TypeError("Unsupported image type")
return output
iface = gr.Interface(
fn=remove_background,
inputs=gr.Image(label="Upload Image"),
outputs=gr.Image(label="Output Image")
)
if __name__ == "__main__":
iface.launch()
|