mnaylor
/

psychbert-cased

Model card Files Files and versions

Mitch Naylor commited on Sep 2, 2021

Commit

735ebaf

·

1 Parent(s): 9fdc17d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -1,7 +1,7 @@
 # PsychBERT
 This domain adapted language model is pretrained from the `bert-base-cased` checkpoint on masked language modeling, using a dataset of ~40,000 PubMed papers in the domain of psychology, psychiatry, mental health, and behavioral health; as well as a dastaset of roughly 200,000 social media conversations about mental health. This work is submitted as an entry for BIBM 2021.
-**Note**: the widget to the right does not work with Flax models. In order to use the model, please pull it into a Python session as follows:
 ```
 from transformers import FlaxAutoModelForMaskedLM, AutoModelForMaskedLM

 # PsychBERT
 This domain adapted language model is pretrained from the `bert-base-cased` checkpoint on masked language modeling, using a dataset of ~40,000 PubMed papers in the domain of psychology, psychiatry, mental health, and behavioral health; as well as a dastaset of roughly 200,000 social media conversations about mental health. This work is submitted as an entry for BIBM 2021.
+**Note**: the token-prediction widget on this page does not work with Flax models. In order to use the model, please pull it into a Python session as follows:
 ```
 from transformers import FlaxAutoModelForMaskedLM, AutoModelForMaskedLM