onnxruntime/download/
language.rs

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
//! Module defining natural language models available to download.
//!
//! See [https://github.com/onnx/models#machine_comprehension](https://github.com/onnx/models#machine_comprehension).

use super::ModelUrl;

pub mod machine_comprehension;

// Re-exports
pub use machine_comprehension::MachineComprehension;

/// Natural language models
#[derive(Debug, Clone)]
pub enum Language {
    /// Machine comprehension
    MachineComprehension(MachineComprehension),
}

impl ModelUrl for Language {
    fn fetch_url(&self) -> &'static str {
        match self {
            Language::MachineComprehension(variant) => variant.fetch_url(),
        }
    }
}