README.md

---
pipeline_tag: text-classification
tags:
- phishing-detection
- mobilebert
- emails
license: apache-2.0
---

# MobileBERT for Phishing Email Detection

This model is a fine-tuned MobileBERT for binary phishing-vs-legitimate email classification.

## Usage

Python (transformers):

```python
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

tok = AutoTokenizer.from_pretrained("Perth0603/phishing-email-mobilebert")
mdl = AutoModelForSequenceClassification.from_pretrained("Perth0603/phishing-email-mobilebert")
inputs = tok(["Win an iPhone! Click here"], return_tensors="pt")
with torch.no_grad():
    pred = mdl(**inputs).logits.softmax(-1)
print(pred)
```

HF Inference API (public models only):

```bash
curl -X POST https://api-inference.huggingface.co/models/Perth0603/phishing-email-mobilebert   -H "Content-Type: application/json"   -d '{"inputs": "Win an iPhone! Click here"}'
```