Update README.md
Browse files
README.md
CHANGED
|
@@ -174,15 +174,5 @@ predicted_class_id = torch.argmax(logits, dim=1).item()
|
|
| 174 |
label_map = model.config.id2label
|
| 175 |
print(f"Predicted label: {label_map[predicted_class_id]}")
|
| 176 |
---
|
| 177 |
-
---
|
| 178 |
-
|
| 179 |
-
```bibtex
|
| 180 |
-
@misc{distilbert-mentalhealth,
|
| 181 |
-
author = {Dileep Reddy Suram},
|
| 182 |
-
title = {DistilBERT Mental Health Classifier},
|
| 183 |
-
howpublished = {\url{https://huggingface.co/dsuram/distilbert-mentalhealth-classifier}},
|
| 184 |
-
year = {2025}
|
| 185 |
-
}
|
| 186 |
|
| 187 |
-
---
|
| 188 |
|
|
|
|
| 174 |
label_map = model.config.id2label
|
| 175 |
print(f"Predicted label: {label_map[predicted_class_id]}")
|
| 176 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 177 |
|
|
|
|
| 178 |
|