import gradio as gr

from inference import inference
from prepare_samples import prepare_samples

TITLE = "BBS-Net Demo"
DESCRIPTION = "Gradio demo for BBS-Net: RGB-D salient object detection with a bifurcated backbone strategy network."
examples = prepare_samples()

demo = gr.Interface(
    fn=inference,
    inputs=[
        gr.inputs.Image(label="RGB", type="pil"),
        gr.inputs.Image(label="Depth", type="pil"),
    ],
    outputs=[
        gr.outputs.Image(label="Prediction", type="pil"),
    ],
    title=TITLE,
    examples=examples,
    description=DESCRIPTION,
)


if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0")