File size: 345 Bytes
95a864c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from transformers import pipeline
import gradio as gr

pipe = pipeline(
    "image-segmentation",
    model="briaai/RMBG-1.4",
    revision="refs/pr/9",
    trust_remote_code=True,
)


def predict(image_path):
    pipe(image_path, out_name="myout.png")
    return "myout.png"


gr.Interface(predict, gr.Image(type="filepath"), "image").launch()