| | 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() |
| |
|
