akuna-infer 0.1.0

Magika file-type detection with Burn
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

//! File-type detection using Magika and Burn.
//!
//! # Example
//!
//! ```rust,no_run
//! use akuna_infer::Session;
//! use burn_wgpu::{Wgpu, WgpuDevice};
//!
//! fn main() -> Result<(), Box<dyn std::error::Error>> {
//!     let device = WgpuDevice::default();
//!     let mut session = Session::<Wgpu>::new(&device)?;
//!
//!     let detected = session.identify_content_sync(b"fn main() { println!(\"hi\"); }")?;
//!     println!("{} {}", detected.info().label, detected.info().mime_type);
//!
//!     Ok(())
//! }
//! ```

mod config;
mod content {
    pub use crate::vendor::content::*;
}
mod detection;
mod file;
mod model;
mod preprocess;
mod session;
mod vendor;

pub use config::ModelConfig;
pub use detection::{Detection, RankedAlternative};
pub use file::{FileType, InferredType, OverwriteReason, TypeInfo};
pub use model::MagikaInferenceError as Error;
pub use model::{MagikaInferenceError, MagikaModel};
pub use preprocess::preprocess_bytes;
pub use session::Session;
pub use vendor::content::{ContentType, MODEL_MAJOR_VERSION, MODEL_NAME};