dsuram commited on
Commit
3595229
·
verified ·
1 Parent(s): 2eb8618

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -10
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