---
license: apache-2.0
language:
  - en
metrics:
  - accuracy
base_model:
  - distilbert-base-uncased
pipeline_tag: text-classification
library_name: transformers
tags:
  - text-classification
  - news
  - distilbert
  - huggingface
  - ag_news
---

# 📚 DistilBERT News Classifier by Dheeraj

A text classification model fine-tuned on news headlines to classify them into one of four categories:

- `0 → Sports`  
- `1 → Business`  
- `2 → Tech`  
- `3 → Science`  

## 🧠 Model Info

This model is built using [DistilBERT](https://huggingface.co/distilbert-base-uncased) and fine-tuned using Hugging Face's `Trainer` API.  
It is trained on a small dataset of sample headlines, inspired by AG News, for demonstration and educational purposes.

## 🚀 Example Usage

```python
from transformers import pipeline

classifier = pipeline("text-classification", model="Dheeraj3103/distilbert-news-classifier-dheeraj")
result = classifier("NASA finds evidence of water on Mars.")
print(result)