Update README.md

README.md

@@ -6,9 +6,10 @@ language: ar
 ## Finetuned Arabic GPT-2 demo
 
 This is a small GPT-2 model, originally trained on Arabic Wikipedia circa September 2020 ,
-finetuned on part of University of British Columbia's Arabic Dialect dataset https://github.com/UBC-NLP/aoc_id
+finetuned on dialect datasets DART (Qatar University), aoc_id (University of British Columbia / NLP)
+and arabic_dialect_annotation (Johns Hopkins University / LREC)
 
-You can use special tokens to set four dialects: `[EGYPTIAN]`, `[GULF]`, `[LEVANTINE]`, and `[MSA]`
+You can use special tokens to prompt five dialects: `[EGYPTIAN]`, `[GULF]`, `[LEVANTINE]`, '[MAGHREBI]', and `[MSA]`
 
 ```
 from simpletransformers.language_generation import LanguageGenerationModel