Ammar-alhaj-ali commited on
Commit
700dc83
1 Parent(s): eb6e548

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -1
README.md CHANGED
@@ -13,4 +13,15 @@ tags:
13
  ---
14
 
15
 
16
- **arabic-MARBERT-dialect-identification-city Model** is a dialect identification model that was built by fine-tuning the [MARBERT](https://huggingface.co/UBC-NLP/MARBERT) model. For the fine-tuning, I used [MADAR Corpus 26 dataset](https://camel.abudhabi.nyu.edu/madar-shared-task-2019/), which includes 26 labels(cities).
 
 
 
 
 
 
 
 
 
 
 
 
13
  ---
14
 
15
 
16
+ **arabic-MARBERT-dialect-identification-city Model** is a dialect identification model that was built by fine-tuning the [MARBERT](https://huggingface.co/UBC-NLP/MARBERT) model. For the fine-tuning, I used [MADAR Corpus 26 dataset](https://camel.abudhabi.nyu.edu/madar-shared-task-2019/), which includes 26 labels(cities).
17
+
18
+
19
+ #### How to use
20
+ To use the model with a transformers pipeline:
21
+ ```python
22
+ >>> from transformers import pipeline
23
+ >>> model= pipeline('text-classification', model='Ammar-alhaj-ali/arabic-MARBERT-dialect-identification-city')
24
+ >>> sentences = ['ناطرين البرنامج', 'اكلنا هوا بهل شروة']
25
+ >>> model(sentences)
26
+ [{'label': 'Beirut', 'score': 0.9731963276863098},
27
+ {'label': 'Aleppo', 'score': 0.4592577815055847}]