"""Hugging Face ZeroGPU entry point of the CLICK3R demo."""

import spaces  # Must be imported before torch on ZeroGPU.
from huggingface_hub import hf_hub_download

from click3r.demo.app import create_interface, load_click3r
from dust3r.model import AsymmetricCroCo3DStereo

OUTPUT = "/tmp/click3r_demo"

model, img_size = load_click3r(
    hf_hub_download("uva-cv-lab/Click3R", "click3r.pth"), "cuda"
)
original = (
    AsymmetricCroCo3DStereo.from_pretrained(
        "naver/DUSt3R_ViTLarge_BaseDecoder_224_linear"
    )
    .to("cuda")
    .eval()
)

interface = create_interface(
    model,
    "cuda",
    OUTPUT,
    "click3r.pth",
    img_size=img_size,
    original_model=original,
    gpu_decorator=spaces.GPU(duration=30),
)

if __name__ == "__main__":
    interface.queue(default_concurrency_limit=1, max_size=16).launch(
        allowed_paths=[OUTPUT]
    )