Downloads · 30 days
28
2% of all-time downloads
nllb-ensi-v1.6
zaanind/nllb-ensi-v1.6
nllb-ensi-v1.6 is a translation model from zaanind. Use it when you need text moved from one language to another. It is set up for transformers. The card lists the license as cc-by-nc-4.0.
It is a NLLB-200-600M model fine-tuned for translating between englih and sinhala languages
All-time downloads
1.3K
Public
Repo size
24.6 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin2.5 GB · 99%
At a glance
- Task
- Translation
- Library
- transformers
- License
- cc-by-nc-4.0
- Model type
- m2m_100
- Access
- Public
- Created
- Dec 13, 2023
- Updated
- Jul 16, 2024
- SHA
- e200592b
Try a prompt
Model card
From the Hugging Face model README
What Is This?
It is a NLLB-200-600M model fine-tuned for translating between englih and sinhala languages
Training & Inference Codes - https://github.com/zaanind/NLLB-200-Sinhala
Try It
How to use the model:
1.Install necessary libraries
pip install requests sentencepiece transformers==4.33 sacremoses sacrebleu
2.Translate!
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
import torch
model_name = "zaanind/nllb-ensi-v1-tuning" #download model
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
def dotranslate(text): #define function for generation
inputs = tokenizer(text, return_tensors="pt")
translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id["sin_Sinh"])
out = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
return out
dotranslate("hello how are you?") #translate
Contact - https://t.me/zaanind
- Task
- Translation
- Library
- transformers
- Type
- m2m_100
- License
- cc-by-nc-4.0
- Languages
- en, si
- Created
- Dec 13, 2023
- Updated
- Jul 16, 2024