use usls::{models::SVTR, DataLoader, Options};

fn main() -> Result<(), Box<dyn std::error::Error>> {
    // build model
    let options = Options::default()
        .with_ixx(0, 0, (1, 2, 8).into())
        .with_ixx(0, 2, (320, 960, 1600).into())
        .with_ixx(0, 3, (320, 960, 1600).into())
        .with_confs(&[0.2])
        .with_vocab("svtr/ppocr_rec_vocab.txt")?
        .with_model("svtr/ppocr-v4-svtr-ch-dyn.onnx")?;
    let mut model = SVTR::new(options)?;

    // load images
    let dl = DataLoader::new("./examples/svtr/images")?.build()?;

    // run
    for (xs, paths) in dl {
        let ys = model.run(&xs)?;
        println!("{paths:?}: {:?}", ys[0].texts())
    }

    Ok(())
}