Struct rust_bert::pipelines::translation::TranslationModel

pub struct TranslationModel { /* fields omitted */ }

TranslationModel to perform translation

Implementations

impl TranslationModel

pub fn new(translation_config: TranslationConfig) -> Fallible<TranslationModel>

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::{TranslationModel, TranslationConfig, Language};
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(&mut self, texts: &[&str]) -> Vec<String>

Translates texts provided

Arguments

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

Returns

• Vec<String> Translated texts

Example

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

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

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

let output = model.translate(&input);