--- license: cc-by-sa-4.0 language: - tig base_model: - facebook/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 ```bash pip install transformers torch
```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 ክል ዶል ኢገብእ መስል እስከ ይከለስ
ኩሉ ለወቅት እግል ልግበእ ኢቀድር መስል
እት እድንየ እግል ትርኤ ተሐዜዮ ተቅዪር ግበእ
እት እድንየ እግል ትርኤ ተሐዜ ለቀየረት ግበእ
እርትራ እት ቀር ቀር አፍሪቀ ደውለት ኮይነ እት ምውዳቅ ጸሓይ ምስ ሱዳን እት ቅብለት ምስ አቶብየ ትትረከብ
እት ቅብለት ምፍጋር ጸሓይ ምስል ጅቡቲ ወ እት ቅብለት ምፍጋር ጸሓይ ቀየሕ ደዋው ብዞሕ ተ