--- 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
ክል ዶል ኢገብእ መስል እስከ ይከለስ