BeitTigreAI/tigre-nllb-200-3.3B

Tigre Translation with NLLB Model

This project demonstrates how to use a fine-tuned NLLB (No Language Left Behind) model to translate text into Tigre (tig_Ethi), using Hugging Face Transformers. The model used is tigre-nllb-200-distilled-600M, trained specifically for high-quality Tigre translation.

---

🚀 Features

• Translate text from various languages (e.g., English, Arabic) into Tigre.
• Support for multiple source languages via src_lang parameter.
• GPU acceleration (if available).
• Easy-to-use translation function.

---

📦 Setup

Dependencies

pip install transformers torch


<pre>
```python
MODEL_PATH = "BeitTigreAI/tigre-nllb-200-3.3B"

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
import torch

tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_PATH)
device = "cuda" if torch.cuda.is_available() else "cpu"

def tranlateThis(theSentence, src_lang = "eng_Latn", tgt_lang = "tig_Ethi"):

    tokenizer.src_lang = src_lang
    inputs = tokenizer(theSentence, return_tensors="pt", padding=True, truncation=True).to(device)

    translated_tokens = model.generate(
        **inputs,
        forced_bos_token_id=tokenizer.convert_tokens_to_ids(tgt_lang),
        num_beams=5,
        max_length=100,
    )
    translated_sentence = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]

    return translated_sentence

print ( tranlateThis("It always seems impossible until it's done",  "eng_Latn", "tig_Ethi"))
print ( tranlateThis("يبدو الأمر مستحيلاً دائماً حتى يتم إنجازه.", "eng_Latn", "tig_Ethi"))
print("\n")
print ( tranlateThis("Be the change that you wish to see in the world", "eng_Latn", "tig_Ethi"))
print ( tranlateThis("كن التغيير الذي ترغب في رؤيته في العالم", "eng_Latn", "tig_Ethi"))
print("\n")
print ( tranlateThis("ኤርትራ ኣብ ቀርኒ ኣፍሪቃ እትርከብ ሃገር ኮይና፡ ብምዕራብ ምስ ሱዳን፡ ብደቡብ ምስ ኢትዮጵያ", "tir_Ethi", "tig_Ethi"))
print ( tranlateThis("ብደቡባዊ ምብራቕ ምስ ጅቡቲ፡ ብሰሜናዊ ምብራቕ ድማ ምስ ቀይሕ ባሕሪ እትዳወብ ሃገር እያ።", "tir_Ethi", "tig_Ethi"))

📝 Translation

ክል ዶል ኢገብእ መስል እስከ ይከለስ
ኩሉ ለወቅት እግል ልግበእ ኢቀድር መስል

እት እድንየ እግል ትርኤ ተሐዜዮ ተቅዪር ግበእ
እት እድንየ እግል ትርኤ ተሐዜ ለቀየረት ግበእ

እርትራ እት ቀር ቀር አፍሪቀ ደውለት ኮይነ እት ምውዳቅ ጸሓይ ምስ ሱዳን እት ቅብለት ምስ አቶብየ ትትረከብ
እት ቅብለት ምፍጋር ጸሓይ ምስል ጅቡቲ ወ እት ቅብለት ምፍጋር ጸሓይ ቀየሕ ደዋው ብዞሕ ተ