init

app.py

@@ -0,0 +1,23 @@
+import gradio as gr
+from transparent_background import Remover
+
+# Load model
+remover = Remover()  # default setting
+remover = Remover(mode='fast', jit=True, device='cpu', ckpt='latest.pth')  # custom setting
+remover = Remover(mode='base-nightly')  # nightly release checkpoint
+
+
+def run(inp_image, sod_type):
+    return remover.process(inp_image.convert("RGB"), type=sod_type)
+
+
+with gr.Blocks() as demo:
+    with gr.Row():
+        inp_image = gr.Image(type="pil")
+        res_image = gr.Image(type="pil")
+    sod_type = gr.Radio(["rgba", "map", "green", "white", "blur", "overlay"], value="rgba")
+    submit_btn = gr.Button(value="Submit")
+    submit_btn.click(run, inputs=[inp_image, sod_type], outputs=[res_image])
+
+if __name__ == "__main__":
+    demo.launch(debug=False, server_name="0.0.0.0", server_port=8189)

latest.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86b896ce80988f5cae1a401746e621c8389e177e03c667e2ad493c1f84b8dbd4
+size 367520613

requirements.txt

@@ -0,0 +1,8 @@
+torch==2.1.2
+torchvision==0.16.2
+opencv-python==4.9.0.80
+timm==0.9.12
+tqdm==4.66.1
+kornia==0.7.1
+gdown==4.7.1
+transparent-background==1.2.12