Update README.md
Browse files
README.md
CHANGED
|
@@ -3,4 +3,11 @@ metrics:
|
|
| 3 |
- accuracy
|
| 4 |
base_model:
|
| 5 |
- unitary/toxic-bert
|
| 6 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
- accuracy
|
| 4 |
base_model:
|
| 5 |
- unitary/toxic-bert
|
| 6 |
+
---
|
| 7 |
+
Use Model
|
| 8 |
+
```bash
|
| 9 |
+
from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
|
| 10 |
+
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 11 |
+
identity_model = AutoModelForSequenceClassification.from_pretrained("Mridul2003/identity-hate-detector").to(device)
|
| 12 |
+
identity_tokenizer = AutoTokenizer.from_pretrained("Mridul2003/identity-hate-detector")
|
| 13 |
+
```
|