Update README.md
Browse files
README.md
CHANGED
@@ -8,4 +8,29 @@ tags:
|
|
8 |
widget:
|
9 |
- text: "2032 ഒളിമ്പിക്സിന് ബ്രിസ്ബെയ്ന് വേദിയാകും; ഗെയിംസിന് വേദിയാകുന്ന മൂന്നാമത്തെ ഓസ്ട്രേലിയന് നഗരം"
|
10 |
---
|
11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
8 |
widget:
|
9 |
- text: "2032 ഒളിമ്പിക്സിന് ബ്രിസ്ബെയ്ന് വേദിയാകും; ഗെയിംസിന് വേദിയാകുന്ന മൂന്നാമത്തെ ഓസ്ട്രേലിയന് നഗരം"
|
10 |
---
|
11 |
+
## Malayalam news classifier
|
12 |
+
|
13 |
+
### Overview
|
14 |
+
|
15 |
+
This model is trained on top of [MalayalamBert](https://huggingface.co/eliasedwin7/MalayalamBERT) for the task of classifying malayalam news headlines. Presently, the following news categories are supported:
|
16 |
+
|
17 |
+
* Business
|
18 |
+
* Sports
|
19 |
+
* Entertainment
|
20 |
+
|
21 |
+
### Dataset
|
22 |
+
|
23 |
+
The dataset used for training this model can be found [here](https://www.kaggle.com/disisbig/malyalam-news-dataset).
|
24 |
+
|
25 |
+
### Using the model with HF pipeline
|
26 |
+
|
27 |
+
```python
|
28 |
+
from transformers import pipeline
|
29 |
+
|
30 |
+
news_headline = "ക്രിപ്റ്റോ ഇടപാടുകളുടെ വിവരങ്ങൾ ആവശ്യപ്പെട്ട് ആദായനികുതി വകുപ്പ് നോട്ടീസയച്ചു"
|
31 |
+
model = pipeline(task="text-classification", model="bipin/malayalam-news-classifier")
|
32 |
+
|
33 |
+
model(news_headline)
|
34 |
+
# Output
|
35 |
+
# [{'label': 'business', 'score': 0.9979357123374939}]
|
36 |
+
```
|