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 |
|