Update README.md
Browse files
README.md
CHANGED
@@ -142,15 +142,6 @@ Developed by **Dileep Reddy Suram**
|
|
142 |
|
143 |
If you use this model, please cite:
|
144 |
|
145 |
-
```bibtex
|
146 |
-
@misc{distilbert-mentalhealth,
|
147 |
-
author = {Dileep Reddy Suram},
|
148 |
-
title = {DistilBERT Mental Health Classifier},
|
149 |
-
howpublished = {\url{https://huggingface.co/dsuram/distilbert-mentalhealth-classifier}},
|
150 |
-
year = {2025}
|
151 |
-
}
|
152 |
-
|
153 |
-
---
|
154 |
# 📦 How to Use (Quick Start)
|
155 |
|
156 |
```python
|
@@ -184,3 +175,14 @@ label_map = model.config.id2label
|
|
184 |
print(f"Predicted label: {label_map[predicted_class_id]}")
|
185 |
---
|
186 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
142 |
|
143 |
If you use this model, please cite:
|
144 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
145 |
# 📦 How to Use (Quick Start)
|
146 |
|
147 |
```python
|
|
|
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 |
+
|