Views
No views yet
transformers library. Here's an example:1from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
2
3# Load the pre-trained model and tokenizer
4model_name = 'echarif/mBART_for_darija_transaltion'
5model = MBartForConditionalGeneration.from_pretrained(model_name)
6tokenizer = MBart50TokenizerFast.from_pretrained(model_name)
7
8# Prepare your input text (Moroccan Darija)
9input_text = "insert your Moroccan Darija sentence here"
10
11# Tokenize the input text
12inputs = tokenizer(input_text, return_tensors="pt", padding=True)
13
14# Generate the translated output
15translated_tokens = model.generate(**inputs)
16translated_text = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
17
18print(f"Translated Text: {translated_text}")