import gradio as gr
from rembg import rm

def rem_bg(input):
    if input !=None:
        #pass
        result=rm(input)
    else:
        pass
    return result

with gr.Blocks() as myface:
    gr.Markdown("""<center><h1>Fast Background Remover</h1><center>""")
    with gr.Accordion("Details",open=False):
        gr.Markdown("""Fast Background Removal by: <a href='https://github.com/danielgatis/rembg' target='_blank'>Github Repo: danielgatis/rembg</a>""")
    with gr.Row():
        in_win=gr.Pil(label="Input", type="filepath", interactive=True, batch=True, max_batch_size=20)
        out_win=gr.Pil(label="Output")

    in_win.change(rem_bg,in_win,out_win)
myface.queue(concurrency_count=120)
myface.launch()