bipin commited on
Commit
7c1673b
1 Parent(s): 0a34ea4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -1
README.md CHANGED
@@ -8,4 +8,29 @@ tags:
8
  widget:
9
  - text: "2032 ഒളിമ്പിക്‌സിന് ബ്രിസ്‌ബെയ്ന്‍ വേദിയാകും; ഗെയിംസിന് വേദിയാകുന്ന മൂന്നാമത്തെ ഓസ്‌ട്രേലിയന്‍ നഗരം"
10
  ---
11
- # News classifier for malayalam language
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ ```