Downloads · 30 days
191
0% of all-time downloads
Ammar-alhaj-ali/arabic-MARBERT-dialect-identification-city
arabic-MARBERT-dialect-identification-city is a text classification model from Ammar-alhaj-ali. Use it when you need a label for a piece of text. It is set up for transformers.
arabic-MARBERT-dialect-identification-city Model is a dialect identification model that was built by fine-tuning the MARBERT model. For the fine-tuning, I used MADAR Corpus 26 dataset, which includes 26 labels(cities).
Downloads · 30 days
191
0% of all-time downloads
All-time downloads
190K
Public
Repo size
1.3 GB
Likes
12
Public
Click a slice to open those files.
.bin652 MB · 99%
From the Hugging Face model README
arabic-MARBERT-dialect-identification-city Model is a dialect identification model that was built by fine-tuning the MARBERT model. For the fine-tuning, I used MADAR Corpus 26 dataset, which includes 26 labels(cities).
To use the model with a transformers pipeline:
>>>from transformers import pipeline
>>>model = pipeline('text-classification', model='Ammar-alhaj-ali/arabic-MARBERT-dialect-identification-city')
>>>sentences = ['ناطرين البرنامج', 'اكلنا هوا بهل شروة']
>>>model(sentences)
[{'label': 'Beirut', 'score': 0.9731963276863098},
{'label': 'Aleppo', 'score': 0.4592577815055847}]