from transformers import MarianMTModel, MarianTokenizer

model_name = "cibfaye/french-wolof-marian-fr-to-wo"
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)

def translate(text):
    inputs = tokenizer(text, return_tensors="pt")
    translated_tokens = model.generate(**inputs)
    translation = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)
    return translation

text = "Bonjour, comment ça va ?"
translation = translate(text)
print("Translation:", translation)

cibfaye/marian-fr-to-wo-faulty

MarianMT French to Wolof Model

Model Description

Training Procedure

Evaluation Metrics

Usage

cibfaye/
marian-fr-to-wo-faulty