gallek-m2m100 / README.md
amurienne's picture
Update README.md
246e299 verified
metadata
license: mit
datasets:
  - Bretagne/ofis_publik_br-fr
  - Bretagne/OpenSubtitles_br_fr
language:
  - fr
  - br
base_model:
  - facebook/m2m100_418M
pipeline_tag: translation
library_name: transformers

Gallek

  • A French -> Breton Translation Model called Gallek (meaning "French" in Breton).
  • The current model version reached a BLEU score of 40 on a 20% split of the training set.
  • Only monodirectionally fr->br fine-tuned for now.
  • Training details available on the GweLLM Github repository.

Sample test code:

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline

modelcard = "amurienne/gallek-m2m100"

model = AutoModelForSeq2SeqLM.from_pretrained(modelcard)
tokenizer = AutoTokenizer.from_pretrained(modelcard)

translation_pipeline = pipeline("translation", model=model, tokenizer=tokenizer, src_lang='fr', tgt_lang='br', max_length=512, device="cpu")

french_text = "traduis de français en breton: j'apprends le breton à l'école."

result = translation_pipeline(french_text)
print(result[0]['translation_text'])

Demo is available on the Gallek Space