|
|
import gradio as gr |
|
|
from transformers import pipeline |
|
|
from PIL import Image |
|
|
|
|
|
def remove_background(image): |
|
|
pipe = pipeline("image-segmentation", model="briaai/RMBG-1.4", trust_remote_code=True) |
|
|
pillow_mask = pipe(image, return_mask=True) |
|
|
pillow_image = pipe(image) |
|
|
return pillow_image |
|
|
|
|
|
app = gr.Interface( |
|
|
fn=remove_background, |
|
|
inputs=gr.components.Image(type="pil"), |
|
|
outputs=gr.components.Image(type="pil", format="png"), |
|
|
title="Penghapusan Latar Belakang Gambar", |
|
|
description="Unggah gambar dan lihat latar belakang dihapus secara otomatis. Gambar yang dihasilkan akan berformat PNG." |
|
|
) |
|
|
|
|
|
if __name__ == "__main__": |
|
|
app.launch(share=True) |
