Struct rust_bert::pipelines::translation::TranslationModel

pub struct TranslationModel { /* fields omitted */ }

TranslationModel to perform translation

Implementations

impl TranslationModel

pub fn new(
    translation_config: TranslationConfig
) -> Result<TranslationModel, RustBertError>

Build a new TranslationModel

Arguments

• translation_config - TranslationConfig object containing the resource references (model, vocabulary, configuration), translation options and device placement (CPU/GPU)

Example

use rust_bert::pipelines::translation::{Language, TranslationConfig, TranslationModel};
use tch::Device;

let translation_config =
    TranslationConfig::new(Language::FrenchToEnglish, Device::cuda_if_available());
let mut summarization_model = TranslationModel::new(translation_config)?;

pub fn translate<'a, S>(&self, texts: S) -> Vec<String> where
    S: AsRef<[&'a str]>, 

Translates texts provided

Arguments

• input - &[&str] Array of texts to summarize.

Returns

• Vec<String> Translated texts

Example

use rust_bert::pipelines::generation_utils::LanguageGenerator;
use rust_bert::pipelines::translation::{Language, TranslationConfig, TranslationModel};
use tch::Device;

let translation_config =
    TranslationConfig::new(Language::EnglishToFrench, Device::cuda_if_available());
let model = TranslationModel::new(translation_config)?;

let input = ["This is a sentence to be translated"];

let output = model.translate(&input);