Views
No views yet
1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3tokenizer = AutoTokenizer.from_pretrained("QCRI/AraDiCE-msa-to-egy")
4model = AutoModelForSeq2SeqLM.from_pretrained("QCRI/AraDiCE-msa-to-egy")
5
6article = "من مصلحتك أن ترحل من كازابلانكا لفترة. هناك موقع لفرنسا الحرة بالقرب من برازفيل. قد أسهل لك العبور."
7inputs = tokenizer(article, return_tensors="pt")
8
9translated_tokens = model.generate(
10 **inputs, forced_bos_token_id=tokenizer.convert_tokens_to_ids("arz_Arab"), max_length=30
11)
12translation = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
13print(translation)licenses_by-nc-sa_4.0_legalcode.txt file.@article{mousi2024aradicebenchmarksdialectalcultural,
title={{AraDiCE}: Benchmarks for Dialectal and Cultural Capabilities in LLMs},
author={Basel Mousi and Nadir Durrani and Fatema Ahmad and Md. Arid Hasan and Maram Hasanain and Tameem Kabbani and Fahim Dalvi and Shammur Absar Chowdhury and Firoj Alam},
year={2024},
publisher={arXiv:2409.11404},
url={https://arxiv.org/abs/2409.11404},
}