Views
No views yet
ar)en)transformers library:1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3model_name = "LuigiJoseph/Helsinki-ar-en-ft"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
6
7text = "مرحبا بالعالم, كيف حالك؟"
8inputs = tokenizer.encode(text, return_tensors="pt")
9outputs = model.generate(inputs)
10translation = tokenizer.decode(outputs[0], skip_special_tokens=True)
11
12print(translation) # English translations| Model | BLEU Score |
|---|---|
| Base Model (Helsinki-NLP/opus-mt-ar-en) | 42.85 |
| Fine-Tuned Model (this one) | 80.79 |