RGBD-SOD/bbsnet

1import gradio as gr2 3from inference import inference4from prepare_samples import prepare_samples5 6TITLE = "BBS-Net Demo"7DESCRIPTION = "Gradio demo for BBS-Net: RGB-D salient object detection with a bifurcated backbone strategy network."8examples = prepare_samples()9 10demo = gr.Interface(11    fn=inference,12    inputs=[13        gr.inputs.Image(label="RGB", type="pil"),14        gr.inputs.Image(label="Depth", type="pil"),15    ],16    outputs=[17        gr.outputs.Image(label="Prediction", type="pil"),18    ],19    title=TITLE,20    examples=examples,21    description=DESCRIPTION,22)23 24 25if __name__ == "__main__":26    demo.launch(server_name="0.0.0.0")27