use anyhow::Result;
use usls::{models::DINOv2, DataLoader, Options};

fn main() -> Result<()> {
    tracing_subscriber::fmt()
        .with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
        .with_timer(tracing_subscriber::fmt::time::ChronoLocal::rfc_3339())
        .init();

    // images
    let xs = DataLoader::try_read_n(&["./assets/bus.jpg", "./assets/bus.jpg"])?;

    // model
    let options = Options::dinov2_small().with_batch_size(xs.len()).commit()?;
    let mut model = DINOv2::new(options)?;

    // encode images
    let y = model.encode_images(&xs)?;
    println!("Feat shape: {:?}", y.shape());

    Ok(())
}