# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("bipin/malayalam-news-classifier")
model = AutoModelForSequenceClassification.from_pretrained("bipin/malayalam-news-classifier")Quick Links
Malayalam news classifier
Overview
This model is trained on top of MalayalamBert for the task of classifying malayalam news headlines. Presently, the following news categories are supported:
- Business
- Sports
- Entertainment
Dataset
The dataset used for training this model can be found here.
Using the model with HF pipeline
from transformers import pipeline
news_headline = "ക്രിപ്റ്റോ ഇടപാടുകളുടെ വിവരങ്ങൾ ആവശ്യപ്പെട്ട് ആദായനികുതി വകുപ്പ് നോട്ടീസയച്ചു"
model = pipeline(task="text-classification", model="bipin/malayalam-news-classifier")
model(news_headline)
# Output
# [{'label': 'business', 'score': 0.9979357123374939}]
Contact
For feedback and questions, feel free to contact via twitter @bkrish_
- Downloads last month
- 7
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="bipin/malayalam-news-classifier")