import gradio as gr

from inference import inference
from prepare_samples import prepare_samples

TITLE = "DPT Depth"
DESCRIPTION = "Gradio demo for DPT-Depth"
examples = prepare_samples()

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


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