Ammar-alhaj-ali
commited on
Commit
•
700dc83
1
Parent(s):
eb6e548
Update README.md
Browse files
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}]
|