README.md

@@ -4,7 +4,7 @@ tags:
 - financial-sentiment-analysis
 - sentiment-analysis
 widget:
-- text: "growth is strong and we have plenty of liquidity"
+- text: "Stocks rallied and the British pound gained."
 ---
 
 `FinBERT` is a BERT model pre-trained on financial communication text. The purpose is to enhance financial NLP research and practice. It is trained on the following three financial communication corpus. The total corpora size is 4.9B tokens.
@@ -14,13 +14,10 @@ widget:
 
 More technical details on `FinBERT`: [Click Link](https://github.com/yya518/FinBERT)
 
+Please check out our working paper [`FinBERT—A Deep Learning Approach to Extracting Textual Information`](https://papers.ssrn.com/sol3/papers.cfm?abstract_id=3910214).
+
 This released `finbert-tone` model is the `FinBERT` model fine-tuned on 10,000 manually annotated (positive, negative, neutral) sentences from analyst reports. This model achieves superior performance on financial tone analysis task. If you are simply interested in using `FinBERT` for financial tone analysis, give it a try.
 
-If you use the model in your academic work, please cite the following paper:
-
-Huang, Allen H., Hui Wang, and Yi Yang. "FinBERT: A Large Language Model for Extracting Information from Financial Text." *Contemporary Accounting Research* (2022).
-
-
 # How to use 
 You can use this model with Transformers pipeline for sentiment analysis.
 ```python

config.json

@@ -1,17 +1,18 @@
 {
-  "architectures": [
+"architectures": [
     "BertForSequenceClassification"
   ],
   "id2label": {
-    "0": "Neutral",
+    "0": "neutral",
-    "1": "Positive",
+    "1": "positive",
-    "2": "Negative"
+    "2": "negative"
   },
   "label2id": {
-    "Positive": 1,
+    "positive": 1,
-    "Negative": 2,
+    "negative": 2,
-    "Neutral": 0
+    "neutral": 0
   },
+  "model_type": "bert",
   "attention_probs_dropout_prob": 0.1,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,